Deep Learning/PyTorch
Torch 내장 모듈을 통한 선형회귀모델 만들기
Raphael_07
2020. 7. 14. 18:50
앞선 포스팅에서는 Pytorch 내장 모듈을 사용하지 않고 수작업으로 선형회귀모델을 구현하였다.
하지만, 앞으로는 개발속도의 향상을 위해 내장된 모듈을 사용하여 개발을 편하게 해보자.
1. torch.nn 모델구축을 편하게 할 수 있도록 함.
2. torch.optim 최적화를 편하게 할 수 있도록 함.
위 두가지 모듈을 기반해서 선형회귀모델을 다시 만들어 보겠다.

Pytorch에 내장된 모듈을 기반으로 선형회귀 모델을 만들어도 원하는 결과를 얻을 수 있음을 확인하였다.