Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

how to save a neural network pytorch

Saving:
	torch.save(model, PATH)


Loading: 
	model = torch.load(PATH)
	model.eval()
Source by pytorch.org #
 
PREVIOUS NEXT
Tagged: #save #neural #network #pytorch
ADD COMMENT
Topic
Name
6+8 =