건설용 자갈 암석 종류 분류 AI 경진대회

알고리즘 | 월간 데이콘 | 비전 | 분류 | macro-f1

  • moneyIcon 상금 : 데이스쿨 프로 구독권
  • 799명 마감

 

[Private 11th] vit large patch14 clip224

2025.06.02 15:34 642 조회 language

데이터셋 자체의 불균형을 처리하는 데에 있어서 GPU만 믿어서는 한계가 분명했습니다. 아래 코드에서도 여러 전처리 방식이 구현되어있기는 하지만, 본질적으로 테스트 데이터셋과 유사한 꼴로 학습 데이터를 변형하는 작업은 생각하지 못했네요. 보다 상위권에 위치한 분들의 솔루션을 보고 많이 배워갑니다. 
A100 Is Not All You Need...

코드 자체는 GPU 2장을 동시에 돌리기 위해서 Accelerate을 이용합니다.

코드