flyers / mlsh

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Meta-Learning Shared Hierarchies

Code for Meta-Learning Shared Hierarchies.

Running Experiments
python main.py --task AntBandits-v1 --num_subs 2 --macro_duration 1000 --num_rollouts 2000 --warmup_time 20 --train_time 30 --replay False AntAgent

The MLSH script works on any Gym environment that implements the randomizeCorrect() function. See the envs/ folder for examples of such environments.

About


Languages

Language:Python 100.0%