I am trying the new feature from Pytorch (dev version) which enables loading a Pytorch model in C++ without any Python dependencies. I am using the tracing method:
import torch import torchvision # An instance of your model. model = A UNET MODEL FROM FASTAI which has hooks as required by UNET # An example input you would normally provide to your model's forward() method. example = torch.rand(1, 3, 224, 224) # Use torch.jit.trace to generate a torch.jit.ScriptModule via tracing. traced_script_module = torch.jit.trace(model, example)
However, I got a value error:
ValueError: Modules that have hooks assigned can't be compiled
A brief look at the error trace:
1118 if orig._backward_hooks or orig._forward_hooks or orig._forward_pre_hooks: -> 1119 raise ValueError("Modules that have hooks assigned can't be compiled") 1120 1121 for name, submodule in orig._modules.items():
I am using it for UNET MODEL from FASTAI. I was wondering if anyone else has tried it and would like to share their experiences if it worked. Also, does anyone if this new feature comes with a limitation of only working for very simple models?