So, I’m trying to replicate the tutorial found here 19 but it doesn’t seem to work.
Loading the model then passing the model and the example to the trace function works, but when I try to check the output by doing this: out = traced_script_module(torch.ones(1,3,244,244))
it gives me an error incompatible function arguments. The following argument types are supported
Leaving that aside and moving forward with the tutorial, calling save() on traced_script_module also gives an error : AttributeError: ‘function’ object has no attribute 'save’
I’m running Linux and using Python 3.6. I’ve tried this with Anaconda and also with vanilla python 3.6, none of them seem to work.