issues
search
boostcampaitech7
/
level2-objectdetection-cv-21
level2-objectdetection-cv-21 created by GitHub Classroom
5
stars
2
forks
source link
✨ Add swin tiny
#100
Closed
suhyun6363
closed
1 month ago
suhyun6363
commented
1 month ago
Overview
Coco 데이터셋의 class가 80개인데 우리 데이터의 class는 10개로 굳이 large model로 학습할 필요가 없다고 생각이 듦. transformer 기반 detection 다른 모델 swin tiny로 실험
swin s 모델 epoch 30 mAP 0549 달성, epoch 50으로 설정, lr (0.00002, 0.00009)로 범위 조정해서 mAP50 0.541 달성한 것을 확인, 학습 epoch 50에 7시간 반 소모
wandb 그래프를 통해 빠르게 성능을 찍는 것을 확인하여 epoch 50 > 25로 수정
작은 모델이 학습을 더 빨리 해 빠른 실험/비교를 위해 tiny 모델로 변경
Change Log
swin_tiny_config 파일 생성
epcoh 50 > 25, lr 조정, project명 수정
Issue Tags
Closed | Fixed: #86
See also: #50
minseokheo
commented
1 month ago
옵티마이저는 trainer.py 파일 안에서 확인하실 수 있습니다!
suhyun6363
commented
1 month ago
네 옵티마이저는 AdamW 사용했습니다
Overview
Change Log
Issue Tags