小编M.Z*_*.Z.的帖子

我知道我可以通过torch.save(model.state_dict(), FILE)或保存模型torch.save(model, FILE)。但是它们都没有保存模型的架构。

那么我们如何在 PyTorch 中保存模型的架构，就像在 Tensorflow 中创建.pb文件一样？我想对我的模型应用不同的调整。如果我不能保存模型的架构，我有什么比每次复制整个类定义并创建一个新类更好的方法吗？