2024.01.02 DAY69
YOLO v3에서 train하는 부분을 구현하였다. train 부분은 몇가지 설정해주는 것을 제외하고는 이전에 만든 model, loss를 사용하는 부분이다.
📝 배운 내용
train
- train 순서
- 모델 학습
- 모델의 결과와 GT 값을 비교하여 loss 계산
- loss의 gradient 계산
- weight 갱신 후 gradient 초기화
- learning rate 조정
- 이때 lr도 고정해서 사용하는 것이 아니라 multistepLR을 사용하여 조건에 따라 lr를 감소하면서 학습한다
- model을 사용하기 전
model.train()을 통해 모델을 학습모드로 설정한다 model.eval()을 통해서는 evaluation 모드로 설정한다
