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