본문 바로가기

Deep Learning/PyTorch

Torch 내장 모듈을 통한 선형회귀모델 만들기

앞선 포스팅에서는 Pytorch 내장 모듈을 사용하지 않고 수작업으로 선형회귀모델을 구현하였다.
하지만, 앞으로는 개발속도의 향상을 위해 내장된 모듈을 사용하여 개발을 편하게 해보자.

1. torch.nn 모델구축을 편하게 할 수 있도록 함.
2. torch.optim 최적화를 편하게 할 수 있도록 함.

위 두가지 모듈을 기반해서 선형회귀모델을 다시 만들어 보겠다.

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