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': { 'policy': {
'learning_rate': 3e-4, 'learning_rate': 3e-4,
'learning_rate_decay': 1.0, 'learning_rate_decay': tune.grid_search([0.001, 1.0]),
'hidden_layer_size': tune.grid_search([20, 40]), 'hidden_layer_size': tune.grid_search([10, 20, 40, 80]),
'n_hidden_layers': tune.grid_search([2, 3]), 'n_hidden_layers': tune.grid_search([2, 3, 4]),
'activation':0, 'activation':0,
}, },
'train_epochs': args.epochs, 'train_epochs': args.epochs,