我目前有一个通过torch.jit.load
. 我想获取一些我拥有的数据并在这些权重之上进行训练,但是我不知道如何训练序列化的 torchscript 模型。
问问题
436 次
1 回答
0
事实证明,返回的ScriptModule
确实支持培训:https ://pytorch.org/docs/stable/generated/torch.jit.ScriptModule.html#torch.jit.ScriptModule.train
于 2021-05-19T15:04:57.370 回答