Hi, I have an exported torch script module that looks loosely like:
class Model(torch.nn.Module):
def forward(self, tensor):
// do forward pass
@torch.jit.export
def reinitialize_hidden_buffers(self, tensor):
// reinitializes model parameters
When I load the torchscript module in Python, I can call reinitialize_hidden_buffers, but when the model is exported and loaded into objective c (for deployment on iOS), I get
no member named 'reinitialize_hidden_buffers' in 'torch::jit::script::Module
Is there a way to call this correctly? Or should I just put control flow logic in the forward pass to do it.