無法保存 Tensorflow_probability 模型

Question

我正在嘗試保存下面的張量流模型：

Model: "sequential_117"
_________________________________________________________________
 Layer (type)                Output Shape              Param #   
=================================================================
 fcl_0 (DenseFlipout)        (None, 47)                1175      
                                                                 
 BN_0 (BatchNormalization)   (None, 47)                188       
                                                                 
 fcl_1 (DenseFlipout)        (None, 14)                1330      
                                                                 
 final_layer (DenseFlipout)  (None, 1)                 29        
                                                                 
=================================================================
Total params: 2,722
Trainable params: 2,628
Non-trainable params: 94
_________________________________________________________________

但我面臨這個錯誤：TypeError: cannot pickle 'module' object

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_14328/3501717249.py in <module>
      1 # save model
----> 2 best_model.save("model_BNN")
      

[...]

~\Anaconda3\lib\copy.py in deepcopy(x, memo, _nil)
    159                     reductor = getattr(x, "__reduce_ex__", None)
    160                     if reductor is not None:
--> 161                         rv = reductor(4)
    162                     else:
    163                         reductor = getattr(x, "__reduce__", None)

TypeError: cannot pickle 'module' object

有人知道這個問題嗎？ 那是張量流概率的錯誤嗎？

提前致謝。

Answer 1

model.save中的 model.save 只接受保存對象的名稱，而不是擴展名。 這是因為它將一堆東西保存在具有該名稱的文件夾中。

如果您嘗試保存整個模型， best_model.save("model_BNN")應該適用於您的情況。

在此處查看文檔。 如果你只想保存訓練模型的權重，你可以使用這個