Skip to content

Latest commit

 

History

History
15 lines (10 loc) · 523 Bytes

README.md

File metadata and controls

15 lines (10 loc) · 523 Bytes

pstage_03_KLUE_Relation_extraction

프로젝트 Wrap up report 여기서 확인가능 합니다.

training

  • python train_roberta.py --name=[training_name] --random_seed=[seed]

training K-fold

  • python train_roberta_kfold.py --name=[training_name] --random_seed=[seed]

inference

  • python inference_roberta.py --model_dir=[model_path]
  • ex) python inference_roberta.py --model_dir=./results/checkpoint-500