pytorch save model

1Saving:
2	torch.save(model, PATH)
3
4
5Loading: 
6	model = torch.load(PATH)
7	model.eval()
8    
9A common PyTorch convention is to save models using either a .pt or .pth file extension.

1Saving:
2	torch.save(model, PATH)
3
4
5Loading: 
6	model = torch.load(PATH)
7	model.eval()

1import dill
2
3model_copy=dill.dumps(model)
4torch.save(model_copy,‘model_ignite_original.pt’)
5
6model1 = torch.load(model_name)
7model=dill.loads(model1)