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?
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?