摘要: 矩阵运算相关: torch.mul(a,b) 是矩阵a和b对应位相乘,a和b的维度必须相等,比如a的维度是(1, 2),b的维度是(1, 2),返回的仍是(1, 2)的矩阵 torch.mm(a,b) 是矩阵a和b的 矩阵相乘。比如a的维度是(1, 2),b的维度是(2, 3),返回的就是(1, 3 阅读全文
posted @ 2020-10-20 21:51 不学无墅_NKer 阅读(421) 评论(0) 推荐(0) 编辑
摘要: 错误原因: 训练保存模型时,torch的版本是1.6.0(使用torch.__version__可以查看torch的版本号) 而加载模型时,torch的版本号低于1.6.0 解决方案: If for any reason you want torch.save to use the old form 阅读全文
posted @ 2020-10-20 09:43 不学无墅_NKer 阅读(12121) 评论(0) 推荐(0) 编辑