Is it possible to use torch.jit.trace (or torch.jit.script) to save the model as a .pt ?
· Sign up or log in to comment