摘要：
1. 出错代码行 计算交叉熵是出现异常提示：RuntimeError: multi-target not supported at /opt/conda/conda-bld/pytorch_1549635019666/work/aten/src/THNN/generic/ClassNLLCriter 阅读全文
posted @ 2019-12-10 20:11
闪存第一菜鸡
阅读(16838)
评论(0)
推荐(1)
摘要：
解决方法： Pytorch使用Pickle来处理保存/加载模型，这个问题实际上是Pickle的问题，而不是Pytorch。 解决方法也非常简单，只需显式地导入类定义。即将包含类定义的文件复制粘贴到与要运行的文件同一文件夹下，再 import Class！ 阅读全文
posted @ 2019-12-10 18:19
闪存第一菜鸡
阅读(9173)
评论(0)
推荐(0)
摘要：
pytorch的模型和参数是分开的，可以分别保存或加载模型和参数。 pytorch有两种模型保存方式：一、保存整个神经网络的的结构信息和模型参数信息，save的对象是网络net 二、只保存神经网络的训练模型参数，save的对象是net.state_dict() 对应两种保存模型的方式，pytorch 阅读全文
posted @ 2019-12-10 11:18
闪存第一菜鸡
阅读(1842)
评论(0)
推荐(0)
浙公网安备 33010602011771号