huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updating weight decay

Browse Source
This commit is contained in:
Arec Jamgochian
2022-02-28 03:35:00 -08:00
parent 62c28d0cfa
commit 5ada1cc543
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bc-experiment.py
View File

@@ -162,7 +162,7 @@ if __name__ == '__main__':
}, },
'policy': { 'policy': {
'learning_rate': 3e-4, 'learning_rate': 3e-4,
'learning_rate_decay': tune.grid_search([0.001, 1.0]), 'learning_rate_decay': tune.grid_search([0.999, 1.0]),
'hidden_layer_size': tune.grid_search([10, 20, 40, 80]), 'hidden_layer_size': tune.grid_search([10, 20, 40, 80]),
'n_hidden_layers': tune.grid_search([2, 3, 4]), 'n_hidden_layers': tune.grid_search([2, 3, 4]),
'activation':0, 'activation':0,