TypeError: '_IncompatibleKeys' object is not callable

在使用pytorch編寫代碼時,出現了"TypeError: ‘_IncompatibleKeys’ object is not callable",報錯原因如下:

model = model.load_state_dict(torch.load('model.pt'))

model.load_state_dict(torch.load(‘model.pt’))返回結果不是model類型,正確寫法是:

model.load_state_dict(torch.load(PATH))

調用model.load_state_dict之後即可加載成功,不用單獨再賦變量。

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