相关疑难解决方法(0)

我正在寻找在PyTorch中保存训练模型的替代方法.到目前为止,我找到了两种选择.

1. torch.save()保存模型,torch.load()加载模型.
2. model.state_dict()用于保存训练有素的模型,而model.load_state_dict()用于加载已保存的模型.

我已经讨论过这种讨论,其中建议方法2优于方法1.

我的问题是,为什么第二种方法更受欢迎？是否因为torch.nn模块具有这两个功能而我们被鼓励使用它们？