开发者社区

文档建议反馈控制台

文章/答案/技术大牛

发布

社区首页 >问答首页 >在Torchscript模型之上进行训练

问在Torchscript模型之上进行训练
EN

Stack Overflow用户

提问于 2021-05-18 18:47:42

回答 1查看 236关注 0票数 0

我目前有一个通过torch.jit.load加载的Torchscript模型。我想拿一些我拥有的数据，在这些权重的基础上进行训练，但是我找不到如何训练序列化的torchscript模型。

EN

回答 1

Stack Overflow用户

发布于 2021-05-19 23:04:57

原来返回的ScriptModule确实支持训练：https://pytorch.org/docs/stable/generated/torch.jit.ScriptModule.html#torch.jit.ScriptModule.train

票数 0

EN

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/67584702

复制

相似问题