add grid search

2023-05-17 13:56:24 +07:00 · 2023-05-17 13:56:24 +07:00 · e7e42739e2
commit e7e42739e2
parent dcb11706aa
1 changed files with 1 additions and 1 deletions
--- 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],
        }