Commit eb956d0f authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

vf coeff linear sched increase

parent 19f6ab29
......@@ -33,11 +33,18 @@ def _import_custom_torch_ppg():
from .custom_ppg.ppg import PPGTrainer
return PPGTrainer
def _import_ppo_experimental():
from .ppo_experimental.ppo import PPOTrainer
return PPOTrainer
CUSTOM_ALGORITHMS = {
"custom/CustomRandomAgent": _import_custom_random_agent,
"RandomPolicy": _import_random_policy,
"CustomPPOAgent": _import_custom_ppo_agent,
"CustomTorchPPOAgent": _import_custom_torch_agent,
"CustomTorchPPGAgent": _import_custom_torch_ppg
"CustomTorchPPGAgent": _import_custom_torch_ppg,
"PPOExperimental": _import_ppo_experimental,
}
......@@ -7,7 +7,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-ppg.yaml"
export EXPERIMENT_DEFAULT="experiments/ppo-experimental.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