boostcampaitech6 / level2-klue-nlp-01

level2-klue-nlp-01 created by GitHub Classroom
2 stars 1 forks source link

feat: add loss function LDAM, Label Smoothing #14

Closed jaealways closed 8 months ago

jaealways commented 8 months ago

LDAM loss function Label Smoothing Loss funciton 추가 후 모듈화했습니다.

# loss 
# select between focal, ldam, labsm (else) cross entropy
parser.add_argument(
    '--loss_type', '-lt', required=True, type=str
)

train.py에서 loss funciton 필수로 지정해주셔야 합니다. Imbalanced 사용시 focal gamma=2 디폴트 유지했습니다.