CoolFace
Modelpublic

textattack/roberta-base-imdb

sourceHugging Faceupdated 5y agoView on Hugging Face
2likes971downloads
train_args.json26 linesDownload Raw Back to root
1{2  "model": "roberta-base",3  "dataset": "imdb",4  "dataset_train_split": "train",5  "dataset_dev_split": "test",6  "tb_writer_step": 1000,7  "checkpoint_steps": -1,8  "checkpoint_every_epoch": false,9  "num_train_epochs": 5,10  "early_stopping_epochs": -1,11  "batch_size": 64,12  "max_length": 128,13  "learning_rate": 3e-05,14  "grad_accum_steps": 1,15  "warmup_proportion": 0.1,16  "config_name": "config.json",17  "weights_name": "pytorch_model.bin",18  "enable_wandb": false,19  "output_dir": "/p/qdata/jm8wx/research/text_attacks/textattack/outputs/training/roberta-base-imdb-2020-06-30-17:56/",20  "num_labels": 2,21  "do_regression": false,22  "best_eval_score": 0.91436,23  "best_eval_score_epoch": 2,24  "epochs_since_best_eval_score": 225}26