본문 바로가기

Python/PyTorch

[부스트코스] [파이토치로 시작하는 딥러닝 기초] 딥러닝을 학습시키는 단계

이 글은 파이토치로 시작하는 딥러닝 기초를 수강하고 작성하는 글입니다.

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