Commit 3e0896a7 authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

aug num 3

parent 6b960047
......@@ -55,10 +55,10 @@ procgen-ppo:
scale_reward: 1.0
reset_returns: False
flattened_buffer: True
augment_randint_num: 6 ## Hacky name fix later
augment_randint_num: 3 ## Hacky name fix later
aux_lr: 5.0e-4
value_lr: 1.0e-3
same_lr_everywhere: True
same_lr_everywhere: False
adaptive_gamma: False
final_lr: 5.0e-5
......
......@@ -6,8 +6,8 @@ set -e
#########################################
# export EXPERIMENT_DEFAULT="experiments/impala-baseline.yaml"
export EXPERIMENT_DEFAULT="experiments/custom-torch-ppo.yaml"
# export EXPERIMENT_DEFAULT="experiments/custom-ppg.yaml"
# export EXPERIMENT_DEFAULT="experiments/custom-torch-ppo.yaml"
export EXPERIMENT_DEFAULT="experiments/custom-ppg.yaml"
export EXPERIMENT=${EXPERIMENT:-$EXPERIMENT_DEFAULT}
if [[ -z $AICROWD_IS_GRADING ]]; then
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment