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

adding big tune run options

Browse Source
This commit is contained in:
Arec Jamgochian
2022-02-28 03:04:59 -08:00
parent 6f181a7351
commit 62c28d0cfa
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bc-experiment.py
View File

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