摘要:
矩阵运算相关: 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 阅读全文
摘要:
错误原因: 训练保存模型时,torch的版本是1.6.0(使用torch.__version__可以查看torch的版本号) 而加载模型时,torch的版本号低于1.6.0 解决方案: If for any reason you want torch.save to use the old form 阅读全文