{ "gen_args" : { "model_name_or_path" : "facebook/bart-large", "corpus_dir" : "data/tdata_xfm/", "train_fn" : "train.txt.nowiki", "eval_fn" : "dev.txt.nowiki", "custom_save_checkpoint" : true, "save_tokenizer" : false, "save_at_end" : false, "first_eval_epoch" : 1, "eval_batch_size" : 32, "eval_num_beams" : 1, "max_in_len" : 1024, "max_out_len" : 1024, "max_train_sent_len" : 100, "max_train_graph_len" : 512 }, "model_args": { "no_repeat_ngram_size" : 0 }, "hf_args" : { "output_dir" : "data/model_parse_xfm", "save_strategy" : "epoch", "evaluation_strategy" : "epoch", "fp16" : true, "group_by_length" : true, "do_train" : true, "do_eval" : true, "save_total_limit" : 1, "logging_steps" : 300, "num_train_epochs" : 16, "per_device_train_batch_size" : 8, "gradient_accumulation_steps" : 4, "weight_decay" : 0.004, "learning_rate" : 5e-5, "max_grad_norm" : 1.0, "warmup_steps" : 5200, "seed" : 42 } }