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'))