Hi
Thank you for the great repo.
After training the model i try to save it by:
aux=learn.model
aux=aux.cpu()
traced_cell=torch.jit.script(aux)
traced_cell.save("whatever.pth")
learn.save('whatever')
But unfortunately the kernel dies each time! Is my method incorrect?
Hi
Thank you for the great repo.
After training the model i try to save it by:
aux=learn.model
aux=aux.cpu()
traced_cell=torch.jit.script(aux)
traced_cell.save("whatever.pth")
learn.save('whatever')
But unfortunately the kernel dies each time! Is my method incorrect?