.torch.save与torch.jit.save
1.torch.jit.savetorch.jit.save用来保存编译后的模型,支持跨平台,要注意模型中只能使用pytorch的函数。jit.save支持保存script类型和trace类型的模型,其中script为全量模型,trace为仅保存运行过的路径2.torch.savetorch.save有三种使用场景,参考:https://stackoverflow.com/questions/42703500/best-way-to-save-a-trained-model-in-pytorch1.保存模型
原创
2022-06-21 11:06:39 ·
4121 阅读 ·
1 评论