Skip to content

CORL 2019: Model Based Planning with Energy Based Models

Notifications You must be signed in to change notification settings

yilundu/model_based_planning_ebm

Repository files navigation

Code for Model Based Planning with Energy Based Models

Command for training on continual reacher environment:

python train.py --exp=reacher_mppi --plan_steps=20 --num_plan_steps=80
--datasource=continual_reacher --noise_sim=80 --mppi --g_coeff=1.0 --a_coeff=0.01  --v_coeff=0.1 --save_interval=300 
--num_env=8 

Command for evaluating on continual reacher environment:

python train.py --exp=<exp_name> --plan_steps=20 --num_plan_steps=80 --datasource=continual_reacher
--noise_sim=80 --mppi --g_coeff=10.0 --a_coeff=0.01  --v_coeff=0.1 --save_interval=300   --num_env=1 
--resume_iter=<train iteration> --n_benchmark_exp=1 --rl_train=False --train=False

#TODO Clean up code

About

CORL 2019: Model Based Planning with Energy Based Models

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages