search

boostcampaitech5 / level1_imageclassification-cv-19

level1_imageclassification-cv-19 created by GitHub Classroom
3 stars 3 forks source link

cross_entropy 이용 시 classes 인자로 인한 오류 #1

Closed Bandi120424 closed 1 year ago

Bandi120424 commented 1 year ago

custom_train.py에서 cross_entropy를 사용할 때에는 classes가 없으므로, 오류가 남

Bandi120424 commented 1 year ago

loss.py에 CELoss class 만들어서 해결