pytorch 加載權重 pth 後預測的結果不一樣? 原因是什麼?

最近遇到

pytorch 加載權重後預測的結果不一樣

找一下官方文件

model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
model.eval()

"請記住，在運行推理之前，務必調用model.eval()去設置 dropout 和 batch normalization 層為評估模式。如果不這麽做，可能導致模型推斷結果不一致。"

有可能就是這個原因喔
給大家參考囉

留言板

歡迎留下建議與分享!希望一起交流!感恩!