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(): ...@@ -33,11 +33,18 @@ def _import_custom_torch_ppg():
from .custom_ppg.ppg import PPGTrainer from .custom_ppg.ppg import PPGTrainer
return PPGTrainer return PPGTrainer
def _import_ppo_experimental():
from .ppo_experimental.ppo import PPOTrainer
return PPOTrainer
CUSTOM_ALGORITHMS = { CUSTOM_ALGORITHMS = {
"custom/CustomRandomAgent": _import_custom_random_agent, "custom/CustomRandomAgent": _import_custom_random_agent,
"RandomPolicy": _import_random_policy, "RandomPolicy": _import_random_policy,
"CustomPPOAgent": _import_custom_ppo_agent, "CustomPPOAgent": _import_custom_ppo_agent,
"CustomTorchPPOAgent": _import_custom_torch_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 ...@@ -7,7 +7,8 @@ set -e
# export EXPERIMENT_DEFAULT="experiments/impala-baseline.yaml" # export EXPERIMENT_DEFAULT="experiments/impala-baseline.yaml"
# export EXPERIMENT_DEFAULT="experiments/custom-torch-ppo.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} export EXPERIMENT=${EXPERIMENT:-$EXPERIMENT_DEFAULT}
if [[ -z $AICROWD_IS_GRADING ]]; then 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