textattack
/

bert-base-uncased-yelp-polarity

Text Classification

Inference Endpoints

Model card Files Files and versions Community

system HF staff commited on Jul 9, 2020

Commit

7012c18

•

1 Parent(s): 9773b06

Update train_args.json

Files changed (1) hide show

train_args.json +6 -6

train_args.json CHANGED Viewed

@@ -8,18 +8,18 @@
   "checkpoint_every_epoch": false,
   "num_train_epochs": 5,
   "early_stopping_epochs": -1,
-  "batch_size": 8,
-  "max_length": 512,
   "learning_rate": 5e-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/bert-base-uncased-yelp_polarity-2020-06-30-16:01/",
   "num_labels": 2,
   "do_regression": false,
-  "best_eval_score": 0.5000526315789474,
-  "best_eval_score_epoch": 1,
-  "epochs_since_best_eval_score": 3
 }

   "checkpoint_every_epoch": false,
   "num_train_epochs": 5,
   "early_stopping_epochs": -1,
+  "batch_size": 16,
+  "max_length": 256,
   "learning_rate": 5e-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/bert-base-uncased-yelp_polarity-2020-07-08-10:42/",
   "num_labels": 2,
   "do_regression": false,
+  "best_eval_score": 0.9699473684210527,
+  "best_eval_score_epoch": 4,
+  "epochs_since_best_eval_score": 0
 }