Closed BaekTree closed 2 years ago
No need to change train.py confiturations in hard coding.
Added model configures as well
Moved the documentation of each TrainerAargument element to config_parser.py
config e.g.
{ "TrainingArguments" : { "output_dir":"./results", "save_total_limit":5, "save_steps":500, "num_train_epochs":20, "learning_rate":5e-5, "per_device_train_batch_size":16, "per_device_eval_batch_size":16, "warmup_steps":500, "weight_decay":0.01, "logging_dir":"./logs", "logging_steps":100, "evaluation_strategy":"steps", "eval_steps" : 500, "load_best_model_at_end" : false }, "model" :{ "huggingface": "klue/bert-base" } }
resolve #11
코드잘봤습니다! PR해도 좋을거 같아요! 다른분들도 같이 보고가면 좋을거 같아요~
하이퍼 파라미터 관리하기 쉽게 만들어주셨네요 좋습니다!
No need to change train.py confiturations in hard coding.
Added model configures as well
Moved the documentation of each TrainerAargument element to config_parser.py
config e.g.
resolve #11