add grid search

This commit is contained in:
thanhvc3 2023-05-17 14:13:00 +07:00
parent c050d8f9d9
commit 53443a4026

View File

@ -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],