Oranblock commited on
Commit
52dad95
·
verified ·
1 Parent(s): 53096ab

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +14 -0
config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "bert-base-uncased",
3
+ "text_column": "text",
4
+ "target_column": "label",
5
+ "num_train_epochs": 3,
6
+ "per_device_train_batch_size": 8,
7
+ "per_device_eval_batch_size": 8,
8
+ "warmup_ratio": 0.1,
9
+ "weight_decay": 0.01,
10
+ "learning_rate": 5e-5,
11
+ "fp16": false,
12
+ "push_to_hub": true,
13
+ "hub_model_id": "your_model_name_on_hub"
14
+ }