Create train_config.json

train_config.json

@@ -0,0 +1,13 @@
+{
+  "model_name": "bert-base-uncased",
+  "training_args": {
+    "output_dir": "./results",
+    "num_train_epochs": 3,
+    "per_device_train_batch_size": 8,
+    "per_device_eval_batch_size": 8,
+    "warmup_steps": 500,
+    "weight_decay": 0.01,
+    "logging_dir": "./logs",
+    "logging_steps": 10
+  }
+}