전체 딥러닝 플로우 구현 해보기
데이터 Load 와 전처리
데이터 확인
PyTorch에서는 TF와 이미지를 표현하는데 있어서 차이점이 있음.
- TF - (batch, height, width, channel)
- PyTorch - (batch, channel, height, width)
모델 정의
학습 로직
PyTorch에서는 model을 Training 모드로 변경 후 Training 할 수 있다.
Training mode
Evaluation mode
Evaluation
- autograd engine, 즉 backpropagatin이나 gradient 계산 등을 꺼서 memory usage를 줄이고 속도를 높임
