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

updating weight decay

Browse Source
This commit is contained in:
Arec Jamgochian
2022-02-28 03:35:00 -08:00
parent 62c28d0cfa
commit 5ada1cc543
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bc-experiment.py
View File

@@ -162,7 +162,7 @@ if __name__ == '__main__':
}, },
'policy': { 'policy': {
'learning_rate': 3e-4, '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]), 'hidden_layer_size': tune.grid_search([10, 20, 40, 80]),
'n_hidden_layers': tune.grid_search([2, 3, 4]), 'n_hidden_layers': tune.grid_search([2, 3, 4]),
'activation':0, 'activation':0,
@@ -209,4 +209,4 @@ if __name__ == '__main__':
s = analysis._checkpoints[i]['config']['seed'] s = analysis._checkpoints[i]['config']['seed']
check_dir = analysis._checkpoints[i]['logdir'] check_dir = analysis._checkpoints[i]['logdir']
shutil.copyfile(os.path.join(check_dir,'policy_final.pt'), shutil.copyfile(os.path.join(check_dir,'policy_final.pt'),
os.path.join(savepath, f'policy_seed{s}.pt')) os.path.join(savepath, f'policy_seed{s}.pt'))