brijeshvadi commited on
Commit
f6fe6f2
·
verified ·
1 Parent(s): 00f8f24

Upload training_args.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. training_args.json +13 -0
training_args.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "./results",
3
+ "num_train_epochs": 5,
4
+ "per_device_train_batch_size": 32,
5
+ "per_device_eval_batch_size": 64,
6
+ "learning_rate": 2e-05,
7
+ "weight_decay": 0.01,
8
+ "evaluation_strategy": "epoch",
9
+ "save_strategy": "epoch",
10
+ "load_best_model_at_end": true,
11
+ "metric_for_best_model": "f1",
12
+ "fp16": true
13
+ }