ld_long的博客

torch.save() or model.load_state_dict()有问题？都不是，是我数据集写错了 今天看pytorch 文档看到了参数或模型的保存，就想动手实践一下。然后神奇的问题出现了。 如果我训练完后立刻保存参数到磁盘，然后马上实例化一个新model并载入刚刚保存的参数，再用这个新的model作预测，预测正确率与训练时一样。但是如果我注释掉训练部分。直接实例化新model并载入参数。预测正确率不足1%。于是我就觉得是不是torch.save或者model.load_state_dict用法