From 62c28d0cfa8d6c4d5a6151cbb9785b29091a4dce Mon Sep 17 00:00:00 2001 From: Arec Jamgochian Date: Mon, 28 Feb 2022 03:04:59 -0800 Subject: [PATCH] adding big tune run options --- bc-experiment.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bc-experiment.py b/bc-experiment.py index b50489b..bf03041 100644 --- a/bc-experiment.py +++ b/bc-experiment.py @@ -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,