From 5ada1cc543a007beaba33f13ac790c33f78b54b3 Mon Sep 17 00:00:00 2001 From: Arec Jamgochian Date: Mon, 28 Feb 2022 03:35:00 -0800 Subject: [PATCH] updating weight decay --- bc-experiment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bc-experiment.py b/bc-experiment.py index bf03041..17c5cbe 100644 --- a/bc-experiment.py +++ b/bc-experiment.py @@ -162,7 +162,7 @@ if __name__ == '__main__': }, 'policy': { '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]), 'n_hidden_layers': tune.grid_search([2, 3, 4]), 'activation':0, @@ -209,4 +209,4 @@ if __name__ == '__main__': s = analysis._checkpoints[i]['config']['seed'] check_dir = analysis._checkpoints[i]['logdir'] shutil.copyfile(os.path.join(check_dir,'policy_final.pt'), - os.path.join(savepath, f'policy_seed{s}.pt')) \ No newline at end of file + os.path.join(savepath, f'policy_seed{s}.pt'))