Pytorch 사용법 정리

1. 모델 Save & Load

 

1) 모델 선언

model = Net()

 

2) 모델 저장

torch.save(model.state_dict(), '/model/model.pth')

    - 모델 파라미터들과 저장 경로 설정

 

3) 모델 로드

model = Net()

model.load_state_dict(torch.load('/model/model.pth'))

    - 모델 선언 후, 저정된 모델파일을 불러온다