torch.load和model.load_state_dict区别
加载预训练得到的权重是一个常见的问题，在最近遇到了两个加载权重的方法。分别是torch.load和model.load_state_dict。
torch.load(‘best.pth’)是加载训练好的权重
model.load_state_dict(‘best.pth’)是将训练好的权重加载到我们的模型中
版权声明：本文为ruoruoruo666原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。