Update config.json

change learning rate
Langboat · Dec 29, 2022 · 9e370ee · 9e370ee
1 parent 82743da
commit 9e370ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train/config.json b/train/config.json
@@ -7,7 +7,7 @@
     "test_dataset_path": "data/test_data.json",
     "train_dataset_path": "data/train_data.json",
     "output_dir":"output_dir",         
-    "learning_rate":3e-5,
+    "learning_rate":2e-4,
     "num_train_epochs" : 5,
     "per_device_train_batch_size":4,
     "per_device_eval_batch_size":1,