boostcampaitech5 / level1_imageclassification-cv-07

level1_imageclassification-cv-07 created by GitHub Classroom
1 stars 1 forks source link

CustomAugmentation + ResNet152 추가 #11

Closed 97DongHyeokOH closed 1 year ago

97DongHyeokOH commented 1 year ago

변경 파일

  1. augmentation.py - 성능이 제일 좋게 나온 CustomAugmentaion을 추가했습니다.
  2. model.py - ResNet152를 추가했습니다.

실행

resize=[224,224] 꼭 확인하셔야 됩니다.

--epochs=70 --augmentation=CustomAugmentation --optimizer=Adam --batch_size=32 --model=ResNet152 --lr=0.00002 --imbalanced=True

nstalways commented 1 year ago

따따봉입니다

suhyehye commented 1 year ago

수고하셨습니다 ㅎㅎ

97DongHyeokOH commented 1 year ago

merge 하겠습니다~