{ "model": "albert-base-v2", "dataset": "ag_news", "dataset_train_split": "train", "dataset_dev_split": "test", "tb_writer_step": 1000, "checkpoint_steps": -1, "checkpoint_every_epoch": false, "num_train_epochs": 5, "early_stopping_epochs": -1, "batch_size": 16, "max_length": 128, "learning_rate": 2e-05, "grad_accum_steps": 1, "warmup_proportion": 0.1, "config_name": "config.json", "weights_name": "pytorch_model.bin", "enable_wandb": false, "output_dir": "/p/qdata/jm8wx/research/text_attacks/textattack/outputs/training/albert-base-v2-ag_news-2020-07-01-02:15/", "num_labels": 4, "do_regression": false, "best_eval_score": 0.9471052631578948, "best_eval_score_epoch": 3, "epochs_since_best_eval_score": 1 }