1. Package load
2. 데이터셋 다운로드 및 훈련, 검증, 테스트 데이터셋 구성
3. 하이퍼파라미터 세팅
4. Dataset 및 DataLoader 할당
5. 네트워크 설계
6. train, validation, test 함수 정의
7. 모델 저장 함수 정의
8. 모델 생성 및 Loss function, Optimizer 정의
9. Training
10. 저장된 모델 불러오기 및 test
11. Transfer Learning
1. Package load
2. 데이터셋 다운로드 및 훈련, 검증, 테스트 데이터셋 구성
3. 하이퍼파라미터 세팅
'AI > PyTorch' 카테고리의 다른 글
[PyTorch] torch.optim의 인자 model.parameters() (0) | 2021.06.15 |
---|---|
[PyTorch] torch.nn 제공함수 (0) | 2021.06.15 |
[PyTorch] torch.nn 과 torch.nn.functional (0) | 2021.06.10 |
[PyTorch] nn.MaxPool2d(), nn.BatchNorm2d() (0) | 2021.06.10 |
[PyTorch] nn.ReLU(), F.relu() (0) | 2021.06.10 |