Torchscript call other function rather than forward

user_6643799 0
Viewed 561

When I compile my torch model to torchscript I can make use of the function forward by just calling the torchscript model object model().

But when I want to use another function created on the model I cant call the function. I try to do model.functionName() expecting to call the function functionName but nothing happens.

Any idea on how could I call subfunctions from the model object that are not forward?

Thanks

torch torchscript pytorch
1 Answers

I suppose you should add decorator @torch.jit.export above the method you want to call and convert torchscript model again. After that you can call method by module.run_method(name, args) docs.

user_107223330
Related