파이토치 모델 저장; pytorch model save

---

모델을 저장하는 방법은 2가지가 있다. 

1. 모델의 state dictionary를 저장하는 방법

2. 모델을 통째로 저장하는 방법

 

---

모델의 state dictionary를 저장하는 방법

 

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

 

---

모델을 통째로 저장하는 방법

 

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