Torchscript和Torch.save()直接保存模型的区别？
来源：13-2 PyTorch工程化基础--Torchscript
2227181
2023-11-22
老师，在工程化基础中您提到要将模型序列化，因此引入了Tracing和script两种序列化方法。但Torch.save()除了保存模型参数，也是可以保存整个模型结构和参数的。
那Tracing和script和Torch.save()的区别是什么呢？或者说工程中的序列化和加载，同咱们前面学习中的保存模型和加载模型的主要区别是什么呢？
以上，谢谢。
写回答
1回答
-
会写代码的好厨师
2023-12-06
直接save 的话会保留很多和训练相关的参数，这些参数在推理的时候实际上是用不到的。
00
相似问题