issues
search
boostcampaitech5
/
level1_imageclassification-cv-07
level1_imageclassification-cv-07 created by GitHub Classroom
1
stars
1
forks
source link
[23.04.26] Beit 클래스 수정, CutMix 학습 추가
#12
Closed
nstalways
closed
1 year ago
nstalways
commented
1 year ago
[변경사항]
Beit 클래스에서 마지막 fc layer가 제대로 변경되지 않았던 부분을 수정
[추가사항]
CutMix를 활용해서 모델을 train시키는 train_with_cutmix 함수 추가
mode 인자에 cutmix를 입력하면 정상적으로 동작.
python train.py --mode=cutmix
[변경사항]
[추가사항]
python train.py --mode=cutmix