Resnet50을 Baseline 모델로 사용했습니다. 이진 분류기 때문에 더 큰 모델을 사용하는 것보다는 학습이 빠른 모델을 많은 Fine tuning
을 통해 최적화하는 방법을 사용했습니다.
Optuna를 사용해서 Learning rate와 weight decay 를 최적화하고
loss function으로 Cross entropy를 사용했습니다.
Epoch은 충분히 주었고
Early stopping을 통해 valid loss가 15step이상 증가하지 않을 경우 학습을 멈췄습니다.