From 53443a40263f31d90fd679bc08ffa468cb4f52e8 Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Wed, 17 May 2023 14:13:00 +0700
Subject: [PATCH] add grid search

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 6be60a4..4bb3b06 100644
--- a/main.py
+++ b/main.py
@@ -650,7 +650,7 @@ if __name__ == "__main__":
         )
 
         paramsGrid = {
-            'optimizer__lr': [0.0001, 0.0003, 0.001],
+            'optimizer__lr': [0.0003, 0.001],
             # 'optimizer__weight_decay': [1e-4, 1e-5, 1e-6],
             # 'module__hid_drop': [0.2, 0.5, 0.7],
             # 'module__embed_dim': [300, 400, 500],