[feat] : add optimization loss customizing - Githubissues

boostcampaitech2 / klue-level2-nlp-11

A solution for KLUE Relation Extraction Competition in the 2nd BoostCamp AI Tech by team AI-ESG

4 stars 6 forks source link

[feat] : add optimization loss customizing #52

Closed zerohoon0102 closed 3 years ago

zerohoon0102 commented 3 years ago

작업 내용

loss함수들이 구현되어있는 loss.py 파일 추가
custom Trainer가 구현되어 있는 trainer.py 파일 추가
f1 loss와 focal loss, CE loss 세가지 사용 가능 => 'focal', 'f1', 'CE'로 argparse에서 구분 default값은 'f1'

관련 이슈

Close #51