这个屌丝很懒，什么也没留下！

加载预训练模型——torch.load 与 load_state_dict_torch load的参数映射到模型上

作者：小惠珠哦 | 2024-06-20 09:42:14

torch load的参数映射到模型上

checkpoint = torch.load('./pretrained/Clean_resnet18-200-regular.pth')
           # torch.load加载预训练模型参数
model.load_state_dict(checkpoint['state_dict'])
           # load_state_dict是net的一个方法：把torch.load加载的参数加载到模型中
model.eval()
           # 把模型换成测试模式。因为测试模型的时候不用dropout和BN了
1
2
3
4
5
6

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小惠珠哦/article/detail/739381