Good to hear you fixed it! We changed the API to TorchScript in PyTorch 1.2 to make it easier to use (i.e. you no longer need to change your model to inherit from ScriptModule instead of nn.Module and you don’t need @script_method), you can read more about it here 68. But this is just sugar over the same thing you’re already doing, so if you already have it working you don’t need to change anything.