cngthnh/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add grid search

Browse Source
This commit is contained in:
thanhvc3 2023-05-17 13:54:49 +07:00
parent bbbac51e4c
commit dcb11706aa
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -651,7 +651,7 @@ if __name__ == "__main__":
paramsGrid = { paramsGrid = {
'optimizer__lr': [0.0001, 0.0003, 0.001], 'optimizer__lr': [0.0001, 0.0003, 0.001],
'optimizer__weight_decay': [1e-4, 1e-5, 1e-6], # '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], # 'module__embed_dim': [300, 400, 500],
} }