Torchscript call other function rather than forward

jit
1

Hi!

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

2

torch.jit.export