Saving the model kills the Kernel

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?