关于不同版本torch保存训练参数的问题及其他（SPIN）
发布于2021-07-25 07:45 阅读(1591) 评论(0) 点赞(0) 收藏(2)
由于想把SPIN中渲染框架换成可微渲染，所以使用了pytorch3d，安装pytorch3d需要的torch版本比较高，所以将源代码torch从1.1调至1.7.1，由此也带来了很多bug问题，在此做一个记录。
1.保存的训练参数虽然都是pt文件，但torch1.7会将其设置为压缩包，所以为了方便，可以修改代码，让它保存成原来格式：
改变前：torch.save(checkpoint, checkpoint_filename)
改变后：torch.save(checkpoint, checkpoint_filename,_use_new_zipfile_serialization=False)
2.由于修改框架后，python3.6变为了3.8， 运行显示 “SyntaxWarning: “is” with a literal. Did you mean ""?"
这是因为从 python 3.8 开始，使用 is 和 is not 运算符时，会抛出 SyntaxWarning 语句警告信息，建议分别使用 == 和 != 代替。
故将代码中的is替换为了,警告消失
3.由于修改框架后, scipy的版本从1.0.0升级到了1.4,由于1.2以后scipy不再保留imresize这个功能，同时,scipy升级后，也没有了imrotate功能,查看文档，所以查看文档，得到如下解决方案：
（1）imresize这个函数已在SciPy1.0.0被否决，在SciPy1.3.0中被弃用，所以使用numpy.array(Image.fromarray(arr).resize())来替代
（2）
具体代码如下：
4.torch版本从1.1升级后还会报编码相关的警告，解决方案如下：
import warnings
warnings.filterwarnings(“ignore”, category=UserWarning)
所属网站分类: 技术文章 > 博客
作者：你爱不爱我
链接：http://www.pythonpdf.com/blog/article/491/d60d9aa0dbe5480bf6a9/
来源：编程知识网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容：(最多支持255个字符)