모델 저장 및 로드
torch.save: net.state_dict()를 저장torch.load: load_state_dict로 모델을 로드
1
2
PATH = '저장할 경로'
torch.save(net.state_dict(), PATH)
1
2
net = NeuralNet()
net.load_state_dict(torch.load(PATH))
[PyTorch] Model Save & Load
This post is licensed under CC BY 4.0 by the author.