网站首页 torchvision

• 

  自己搭建resnet18网络并加载torchvision自带权重的操作

  直接搭建网络必须与torchvision自带的网络的权重也就是pth文件的结构、尺寸和变量命名完全一致，否则无法加载权重文件。此时可比较2个字典逐一加载，详见pytorch加载预训练模型与自己模型不匹配的解决方案import to...

  2021-07-14 longtao100 python 558 ℃ 0 评论