모델을 학습시키기 전에,
- 어떤 loss를 쓸 것인지
- learning rate은 몇으로 할 것인지
- optimizer는 무엇으로 할 것인지
- 학습 횟수는 몇 번으로 할 것인지
'AI > PyTorch' 카테고리의 다른 글
[PyTorch] nn.MaxPool2d(), nn.BatchNorm2d() (0) | 2021.06.10 |
---|---|
[PyTorch] nn.ReLU(), F.relu() (0) | 2021.06.10 |
[PyTorch] Dataset and DataLoader 사용법 (0) | 2021.06.01 |
[PyTorch] Multivariable Linear Regression (0) | 2021.06.01 |
[PyTorch] Linear Regression 작성하기 (0) | 2021.06.01 |