keras can‘t pickle _thread.rlock

keras can’t pickle _thread.rlock 网络中使用Lambda层会出现这个异常。 解决办法:checkpointer的save weights only改成true,不保存计算图就行。使用以下代码将.h5转.pb
相关文章
相关标签/搜索