이 글은 파이토치로 시작하는 딥러닝 기초를 수강하고 작성하는 글입니다.
1. 라이브러리 가져오기 (torch, torchvision, matplotlib같은 것들)
2. GPU 사용 설정하고 random value를 위한 seed설정!
3. 학습에 사용되는 parameter 설정! ( learning_rate, training_epochs, batch_size, etc)
4. 데이터셋을 가져오고 (학습에 쓰기 편하게) loader만들기
- 데이터 processing
5. 학습 모델 만들기 (class CNN(torch.nn.Module))
- layer종류, 개수 및 뉴런 개수 설정
6. Loss function (Criterion)을 선택하고 최적화 도구 선택(optimizer)
7. 모델 학습 및 loss check(Crtierion의 output)
- weight와 bias를 조절해서 cost function을 minimize하는 과정
8. 학습된 모델의 성능을 확인한다.
'Python > PyTorch' 카테고리의 다른 글
| [Pytorch] Dataset과 Dataloader (0) | 2021.09.25 |
|---|---|
| [Pytorch] .requires_grad(), torch.no_grad(), model.eval() (0) | 2021.09.06 |
| Custom Dataset 생성 (0) | 2021.08.25 |
| [Pytorch] torch has no "from_numpy" member (0) | 2020.08.02 |
| [Pytorch] Tensor (0) | 2020.07.25 |