Solved by IsaacBerman in post #3 
              
              
                Hi Ivan, 
Thanks for your response. I ended up solving the error. It turned out it was a problem with the tracing – see here: Torch.jit.trace() only works on example input?