5
在PyTorch中保存经过训练的模型的最佳方法?
我一直在寻找其他方法来在PyTorch中保存经过训练的模型。到目前为止,我发现了两种选择。 使用torch.save()保存模型,使用torch.load()加载模型。 model.state_dict()保存训练的模型,model.load_state_dict()加载保存的模型。 我碰到过这种讨论,其中建议方法2优于方法1。 我的问题是,为什么选择第二种方法呢?仅仅是因为torch.nn模块具有这两个功能,我们被鼓励使用它们吗?