无法保存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:无法选择‘模块’对象

---------------------------------------------------------------------------
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

有人对这个问题有什么想法吗？那是tensorflow概率的错误吗？

提前谢谢。

Answer 1

Tensorflow中的model.save只使用保存对象的名称，而不是扩展名。这是因为它将一堆东西保存在一个具有该名称的文件夹中。

如果您想要保存整个模型，那么best_model.save("model_BNN")应该适用于您的情况。

查看文档这里。如果只想保存经过训练的模型的权重，可以使用这