笔者是在写pytorch神经网络保存模型,并对模型使用测试集进行测试的时候出现的这个问题。主要原因是net的class没有更新,导致的这个错误。出现这个错误一般是神经网络的输入和神经网络结构不对应导致的。可以从这方面进行debug。
【error】RuntimeError: size mismatch
最新推荐文章于 2024-05-19 23:31:32 发布
笔者是在写pytorch神经网络保存模型,并对模型使用测试集进行测试的时候出现的这个问题。主要原因是net的class没有更新,导致的这个错误。出现这个错误一般是神经网络的输入和神经网络结构不对应导致的。可以从这方面进行debug。