Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

torch save state dict

torch.save(model.state_dict(), PATH)
model.load_state_dict(torch.load(PATH))
 
PREVIOUS NEXT
Tagged: #torch #save #state #dict
ADD COMMENT
Topic
Name
6+3 =