在 Torchscript 模型之上训练

问题描述

我目前有一个通过 torch.jit.load 加载的 Torchscript 模型。我想获取一些我拥有的数据并在这些权重之上进行训练，但是我不知道如何训练序列化的 torchscript 模型。

解决方法

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