From 9e370ee0fdd2236ebd5518c2cbc410e8d9894c23 Mon Sep 17 00:00:00 2001
From: bling <201711680982@stu.blcu.edu.cn>
Date: Thu, 29 Dec 2022 18:04:02 +0800
Subject: [PATCH] Update config.json

change learning rate
---
 train/config.json | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train/config.json b/train/config.json
index a05d38d..d6b707e 100644
--- 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,