pytorch 신경망 코딩 순서
- 라이브러리 호출
- cpu, gpu 장치 확인
- 데이터셋 가져오기
- 데이터로더에 전달
- 클래스 정의
- 심층 신경망 모델 생성
- 파라미터 정의(learning late, loss function, optimizer)
- 심층 신경망 이용한 모델 학습
import torch.nn
nn.Linear : 선형 변환모듈
들어오는 입력데이터를 선형변환한다.
in_features : 입력데이터 크기
out_features : 출력데이터 크기
bias : default는 true , false 설정시 학습시 레이어가 bias 학습 안함.
'AI > Deep Learning' 카테고리의 다른 글
| audio classification (0) | 2023.02.23 |
|---|---|
| 23 / 01/ 31 AttributeError: '_MultiProcessingDataLoaderIter' object has no attribute 'next' ERROR 해결 (1) | 2023.01.31 |
