Skip to content

Python code snippet – How to save a neural network pytorch?

Saving:
	torch.save(model, PATH)


Loading: 
	model = torch.load(PATH)
	model.eval()
See also  Python code snippet - How to use annoy ?

Leave a Reply

Your email address will not be published. Required fields are marked *

This site uses Akismet to reduce spam. Learn how your comment data is processed.