앞선 포스팅에서는 Pytorch 내장 모듈을 사용하지 않고 수작업으로 선형회귀모델을 구현하였다.
하지만, 앞으로는 개발속도의 향상을 위해 내장된 모듈을 사용하여 개발을 편하게 해보자.
1. torch.nn 모델구축을 편하게 할 수 있도록 함.
2. torch.optim 최적화를 편하게 할 수 있도록 함.
위 두가지 모듈을 기반해서 선형회귀모델을 다시 만들어 보겠다.
Pytorch에 내장된 모듈을 기반으로 선형회귀 모델을 만들어도 원하는 결과를 얻을 수 있음을 확인하였다.
'Deep Learning > PyTorch' 카테고리의 다른 글
다중 분류를 위한 Logistic Regression Model 만들기 (0) | 2020.07.14 |
---|---|
Logistic Regression 구현해보기 (0) | 2020.07.14 |
Tensor를 통한 선형회귀모델 만들기 (0) | 2020.07.14 |
Tensor와 AutoGrad (0) | 2020.07.13 |
텐서(Tensor)의 구조와 기능 (0) | 2020.07.13 |