keras can‘t pickle _thread.rlock

keras can’t pickle _thread.rlock 網絡中使用Lambda層會出現這個異常。 解決辦法:checkpointer的save weights only改成true,不保存計算圖就行。使用以下代碼將.h5轉.pb
相關文章
相關標籤/搜索