CODEWITHSUNDEEP
pytorchtorchscript
user293895
user293895

Reputation: 1527

Train on top of a Torchscript model

I currently have a Torchscript model I load via torch.jit.load. I would like to take some data I have and train on top of these weights, however I cannot find out how to train a serialised torchscript model.

Upvotes: 0

Views: 3290

Answers (1)

user293895
user293895

Reputation: 1527

Turns out that the returned ScriptModule does actually support training: https://pytorch.org/docs/stable/generated/torch.jit.ScriptModule.html#torch.jit.ScriptModule.train

Upvotes: 1

Related Questions