报错:TypeError: can‘t pickle _thread.RLock objects
生活随笔
收集整理的這篇文章主要介紹了
报错:TypeError: can‘t pickle _thread.RLock objects
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
問題:
tensorflow.keras.models.save_model 報錯:TypeError: can’t pickle _thread.RLock objects
參考鏈接:
https://github.com/keras-team/keras/issues/8343
原因分析:
tf.keras.layers.Lambda(function, output_shape=None, mask=None, arguments=None, **kwargs )上面的匿名函數層 Lambda 的第一個參數 function 在定義時用到了類屬性,這樣在模型保存時,便會遇到標題中的報錯。示例及修改方法如下:
# function 函數原始定義中使用了類屬性 self.mean_dim def my_layer(inputs):new_layer = [Dense(self.mean_dim, activation="relu", name="my_layer")]# 將函數定義中的類屬性賦值為共有屬性,便可解決: mean_dim = self.mean_dim def my_layer(inputs):new_layer = [Dense(mean_dim, activation="relu", name="my_layer")]總結
以上是生活随笔為你收集整理的报错:TypeError: can‘t pickle _thread.RLock objects的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 有点牛!谷歌发布AI新模型PaLM2 可
- 下一篇: 2599元 尼康首款Z卡口电动变焦镜头来