网站首页 加载
-
jupyter notebook加载和运行.py文件方式
目录jupyternotebook加载和运行.py文件加载.py文件运行.py文件使用jupyternotebook读取py文件并导出为pdfstep1step2step3step4总结jupyternotebook加载和运行.py...
2023-09-27 longtao100 python 6 ℃ 0 评论 -
pytorch加载预训练模型与自己模型不匹配的解决方案
pytorch中如果自己搭建网络并且加载别人的与训练模型的话,如果模型和参数不严格匹配,就可能会出问题,接下来记录一下我的解决方法。两个有序字典找不同模型的参数和pth文件的参数都是有序字典(OrderedDict),把字典中的键转...
2021-07-16 longtao100 HTML 577 ℃ 0 评论 -
如何解决.cuda()加载用时很长的问题
方法一:pip install --upgrade --force-reinstall http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-c...
2021-07-16 longtao100 python 811 ℃ 0 评论 -
自己搭建resnet18网络并加载torchvision自带权重的操作
直接搭建网络必须与torchvision自带的网络的权重也就是pth文件的结构、尺寸和变量命名完全一致,否则无法加载权重文件。此时可比较2个字典逐一加载,详见pytorch加载预训练模型与自己模型不匹配的解决方案import to...
2021-07-14 longtao100 python 528 ℃ 0 评论
- 开源分类
- 最近发表
- 开源网标签