Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

saving model in pytorch

torch.save(model.state_dict(), PATH)
Source by pytorch.org #
 
PREVIOUS NEXT
Tagged: #saving #model #pytorch
ADD COMMENT
Topic
Name
6+5 =