'tf' is not defined on load_model()

使用keras加載訓練模型進行預測時候報錯:

'tf' is not defined on load_model()

原因是使用了Lambda自定義層,需要參數聲明:

model = load_model('model.model', custom_objects={'tf': tf})

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章