서울 랜드마크 이미지 분류 AI 해커톤

알고리즘 | CV | 분류 | 랜드마크 | Accuracy

상금 : 인증서, 장학금, 스타벅스 기프티콘 등
508명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

이동욱00 | Private 1위 (점수 : 1) | Resnet50 & 10-Fold & Adam

이동욱_00

2022.09.04 18:43 1,644 조회 language

PyTorch의 ResNet50을 기반으로 학습하였습니다.
K-Fold Cross Validation(k는 10으로 설정하였습니다)을 사용하였고,
Optimizer를 SGD에서 Adam으로 바꾸어줬으며,
Learning Rate에 스케줄러를 추가하였습니다(아래 코드 참조).

그 외 Hyperparameter들은 아래 코드에 작성해놓았습니다.
+ 저는 맥북 개발환경을 사용하여 CUDA대신 mps 사용설정을 하였습니다. CUDA 사용설정은 본 대회 개최측에서 제공해주신 Baseline 코드를 봐주시면 감사하겠습니다.

코드