├── .gitignore ├── LICENSE ├── README.md ├── fetch ├── LICENSE ├── __init__.py ├── assets │ ├── LICENSE.md │ ├── bin.xml │ ├── bin_null.xml │ ├── box.xml │ ├── box_bin.xml │ ├── box_bin_drawer.xml │ ├── box_block.xml │ ├── box_no_lid.xml │ ├── drawer.xml │ ├── drawer_bin.xml │ ├── drawer_block.xml │ ├── objects │ │ ├── bin.xml │ │ ├── bin_tall.xml │ │ ├── cabinet.xml │ │ ├── drawer.xml │ │ └── lid.xml │ ├── pick_place.xml │ ├── push.xml │ ├── reach.xml │ ├── robot.xml │ ├── shared.xml │ ├── slide.xml │ ├── stack_two.xml │ ├── stls │ │ ├── base_link_collision.stl │ │ ├── bellows_link_collision.stl │ │ ├── elbow_flex_link_collision.stl │ │ ├── estop_link.stl │ │ ├── forearm_roll_link_collision.stl │ │ ├── gripper_link.stl │ │ ├── head_pan_link_collision.stl │ │ ├── head_tilt_link_collision.stl │ │ ├── l_wheel_link_collision.stl │ │ ├── laser_link.stl │ │ ├── r_wheel_link_collision.stl │ │ ├── shoulder_lift_link_collision.stl │ │ ├── shoulder_pan_link_collision.stl │ │ ├── torso_fixed_link.stl │ │ ├── torso_lift_link_collision.stl │ │ ├── upperarm_roll_link_collision.stl │ │ ├── wrist_flex_link_collision.stl │ │ └── wrist_roll_link_collision.stl │ ├── textures │ │ ├── block.png │ │ └── block_hidden.png │ └── twin_box.xml ├── bin.py ├── box.py ├── common │ ├── __init__.py │ ├── robot_env.py │ ├── rotations.py │ └── utils.py ├── drawer.py ├── fetch_env.py ├── files │ └── bin.py ├── gym_fetch.py ├── mixed_envs │ ├── __init__.py │ ├── box_bin.py │ ├── box_bin_drawer.py │ ├── box_block.py │ ├── drawer_bin.py │ ├── drawer_block.py │ └── twin_box.py ├── no_lid.py ├── scratch_fetch_debug_obs.py ├── stack_two.py └── wrappers.py ├── figures ├── l3p.png └── mbrl.png ├── goal_env ├── __init__.py ├── bitflip.py ├── fourroom.py ├── goal_plane_env.py ├── mountaincar.py ├── mujoco │ ├── __init__.py │ ├── ant.py │ ├── ant_maze_env.py │ ├── assets │ │ ├── ant.xml │ │ └── point.xml │ ├── create_maze_env.py │ ├── maze_env.py │ ├── maze_env_utils.py │ ├── point.py │ └── point_maze_env.py ├── plane.py └── recorder.py ├── rl ├── __init__.py ├── agent │ ├── __init__.py │ ├── core.py │ ├── latent_fetch.py │ ├── latent_planner.py │ ├── normalizer.py │ └── planner.py ├── algo │ ├── __init__.py │ ├── core.py │ └── latent_planner.py ├── launcher_latent.py ├── launcher_latent_fetch.py ├── launcher_latent_robot.py ├── learn │ ├── __init__.py │ ├── core.py │ └── latent_planner.py ├── logger.py ├── main_latent.py ├── main_latent_fetch.py ├── main_latent_robot.py ├── replay │ ├── __init__.py │ ├── core.py │ └── planner.py ├── search │ ├── __init__.py │ └── latent_planner.py └── utils │ ├── __init__.py │ ├── mpi_utils.py │ ├── net_utils.py │ ├── run_utils.py │ ├── torch_utils.py │ └── vec_env │ ├── __init__.py │ ├── base_vec_env.py │ ├── dummy_vec_env.py │ ├── subproc_vec_env.py │ ├── vec_frame_stack.py │ └── vec_normalize.py └── scripts ├── distractor.sh ├── inside_box.sh ├── pick.sh ├── train_antmaze.sh └── train_pointmaze.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/README.md -------------------------------------------------------------------------------- /fetch/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/LICENSE -------------------------------------------------------------------------------- /fetch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/__init__.py -------------------------------------------------------------------------------- /fetch/assets/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/LICENSE.md -------------------------------------------------------------------------------- /fetch/assets/bin.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/bin.xml -------------------------------------------------------------------------------- /fetch/assets/bin_null.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/bin_null.xml -------------------------------------------------------------------------------- /fetch/assets/box.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/box.xml -------------------------------------------------------------------------------- /fetch/assets/box_bin.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/box_bin.xml -------------------------------------------------------------------------------- /fetch/assets/box_bin_drawer.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/box_bin_drawer.xml -------------------------------------------------------------------------------- /fetch/assets/box_block.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/box_block.xml -------------------------------------------------------------------------------- /fetch/assets/box_no_lid.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/box_no_lid.xml -------------------------------------------------------------------------------- /fetch/assets/drawer.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/drawer.xml -------------------------------------------------------------------------------- /fetch/assets/drawer_bin.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/drawer_bin.xml -------------------------------------------------------------------------------- /fetch/assets/drawer_block.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/drawer_block.xml -------------------------------------------------------------------------------- /fetch/assets/objects/bin.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/objects/bin.xml -------------------------------------------------------------------------------- /fetch/assets/objects/bin_tall.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/objects/bin_tall.xml -------------------------------------------------------------------------------- /fetch/assets/objects/cabinet.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/objects/cabinet.xml -------------------------------------------------------------------------------- /fetch/assets/objects/drawer.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/objects/drawer.xml -------------------------------------------------------------------------------- /fetch/assets/objects/lid.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/objects/lid.xml -------------------------------------------------------------------------------- /fetch/assets/pick_place.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/pick_place.xml -------------------------------------------------------------------------------- /fetch/assets/push.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/push.xml -------------------------------------------------------------------------------- /fetch/assets/reach.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/reach.xml -------------------------------------------------------------------------------- /fetch/assets/robot.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/robot.xml -------------------------------------------------------------------------------- /fetch/assets/shared.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/shared.xml -------------------------------------------------------------------------------- /fetch/assets/slide.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/slide.xml -------------------------------------------------------------------------------- /fetch/assets/stack_two.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stack_two.xml -------------------------------------------------------------------------------- /fetch/assets/stls/base_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/base_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/bellows_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/bellows_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/elbow_flex_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/elbow_flex_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/estop_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/estop_link.stl -------------------------------------------------------------------------------- /fetch/assets/stls/forearm_roll_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/forearm_roll_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/gripper_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/gripper_link.stl -------------------------------------------------------------------------------- /fetch/assets/stls/head_pan_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/head_pan_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/head_tilt_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/head_tilt_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/l_wheel_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/l_wheel_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/laser_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/laser_link.stl -------------------------------------------------------------------------------- /fetch/assets/stls/r_wheel_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/r_wheel_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/shoulder_lift_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/shoulder_lift_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/shoulder_pan_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/shoulder_pan_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/torso_fixed_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/torso_fixed_link.stl -------------------------------------------------------------------------------- /fetch/assets/stls/torso_lift_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/torso_lift_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/upperarm_roll_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/upperarm_roll_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/wrist_flex_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/wrist_flex_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/stls/wrist_roll_link_collision.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/stls/wrist_roll_link_collision.stl -------------------------------------------------------------------------------- /fetch/assets/textures/block.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/textures/block.png -------------------------------------------------------------------------------- /fetch/assets/textures/block_hidden.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/textures/block_hidden.png -------------------------------------------------------------------------------- /fetch/assets/twin_box.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/assets/twin_box.xml -------------------------------------------------------------------------------- /fetch/bin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/bin.py -------------------------------------------------------------------------------- /fetch/box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/box.py -------------------------------------------------------------------------------- /fetch/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fetch/common/robot_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/common/robot_env.py -------------------------------------------------------------------------------- /fetch/common/rotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/common/rotations.py -------------------------------------------------------------------------------- /fetch/common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/common/utils.py -------------------------------------------------------------------------------- /fetch/drawer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/drawer.py -------------------------------------------------------------------------------- /fetch/fetch_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/fetch_env.py -------------------------------------------------------------------------------- /fetch/files/bin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/files/bin.py -------------------------------------------------------------------------------- /fetch/gym_fetch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/gym_fetch.py -------------------------------------------------------------------------------- /fetch/mixed_envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fetch/mixed_envs/box_bin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/mixed_envs/box_bin.py -------------------------------------------------------------------------------- /fetch/mixed_envs/box_bin_drawer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/mixed_envs/box_bin_drawer.py -------------------------------------------------------------------------------- /fetch/mixed_envs/box_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/mixed_envs/box_block.py -------------------------------------------------------------------------------- /fetch/mixed_envs/drawer_bin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/mixed_envs/drawer_bin.py -------------------------------------------------------------------------------- /fetch/mixed_envs/drawer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/mixed_envs/drawer_block.py -------------------------------------------------------------------------------- /fetch/mixed_envs/twin_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/mixed_envs/twin_box.py -------------------------------------------------------------------------------- /fetch/no_lid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/no_lid.py -------------------------------------------------------------------------------- /fetch/scratch_fetch_debug_obs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/scratch_fetch_debug_obs.py -------------------------------------------------------------------------------- /fetch/stack_two.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/stack_two.py -------------------------------------------------------------------------------- /fetch/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/fetch/wrappers.py -------------------------------------------------------------------------------- /figures/l3p.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/figures/l3p.png -------------------------------------------------------------------------------- /figures/mbrl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/figures/mbrl.png -------------------------------------------------------------------------------- /goal_env/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/__init__.py -------------------------------------------------------------------------------- /goal_env/bitflip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/bitflip.py -------------------------------------------------------------------------------- /goal_env/fourroom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/fourroom.py -------------------------------------------------------------------------------- /goal_env/goal_plane_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/goal_plane_env.py -------------------------------------------------------------------------------- /goal_env/mountaincar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mountaincar.py -------------------------------------------------------------------------------- /goal_env/mujoco/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/__init__.py -------------------------------------------------------------------------------- /goal_env/mujoco/ant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/ant.py -------------------------------------------------------------------------------- /goal_env/mujoco/ant_maze_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/ant_maze_env.py -------------------------------------------------------------------------------- /goal_env/mujoco/assets/ant.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/assets/ant.xml -------------------------------------------------------------------------------- /goal_env/mujoco/assets/point.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/assets/point.xml -------------------------------------------------------------------------------- /goal_env/mujoco/create_maze_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/create_maze_env.py -------------------------------------------------------------------------------- /goal_env/mujoco/maze_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/maze_env.py -------------------------------------------------------------------------------- /goal_env/mujoco/maze_env_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/maze_env_utils.py -------------------------------------------------------------------------------- /goal_env/mujoco/point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/point.py -------------------------------------------------------------------------------- /goal_env/mujoco/point_maze_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/mujoco/point_maze_env.py -------------------------------------------------------------------------------- /goal_env/plane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/plane.py -------------------------------------------------------------------------------- /goal_env/recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/goal_env/recorder.py -------------------------------------------------------------------------------- /rl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rl/agent/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rl/agent/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/agent/core.py -------------------------------------------------------------------------------- /rl/agent/latent_fetch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/agent/latent_fetch.py -------------------------------------------------------------------------------- /rl/agent/latent_planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/agent/latent_planner.py -------------------------------------------------------------------------------- /rl/agent/normalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/agent/normalizer.py -------------------------------------------------------------------------------- /rl/agent/planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/agent/planner.py -------------------------------------------------------------------------------- /rl/algo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rl/algo/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/algo/core.py -------------------------------------------------------------------------------- /rl/algo/latent_planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/algo/latent_planner.py -------------------------------------------------------------------------------- /rl/launcher_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/launcher_latent.py -------------------------------------------------------------------------------- /rl/launcher_latent_fetch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/launcher_latent_fetch.py -------------------------------------------------------------------------------- /rl/launcher_latent_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/launcher_latent_robot.py -------------------------------------------------------------------------------- /rl/learn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rl/learn/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/learn/core.py -------------------------------------------------------------------------------- /rl/learn/latent_planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/learn/latent_planner.py -------------------------------------------------------------------------------- /rl/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/logger.py -------------------------------------------------------------------------------- /rl/main_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/main_latent.py -------------------------------------------------------------------------------- /rl/main_latent_fetch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/main_latent_fetch.py -------------------------------------------------------------------------------- /rl/main_latent_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/main_latent_robot.py -------------------------------------------------------------------------------- /rl/replay/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rl/replay/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/replay/core.py -------------------------------------------------------------------------------- /rl/replay/planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/replay/planner.py -------------------------------------------------------------------------------- /rl/search/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rl/search/latent_planner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/search/latent_planner.py -------------------------------------------------------------------------------- /rl/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rl/utils/mpi_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/mpi_utils.py -------------------------------------------------------------------------------- /rl/utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/net_utils.py -------------------------------------------------------------------------------- /rl/utils/run_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/run_utils.py -------------------------------------------------------------------------------- /rl/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/torch_utils.py -------------------------------------------------------------------------------- /rl/utils/vec_env/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/vec_env/__init__.py -------------------------------------------------------------------------------- /rl/utils/vec_env/base_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/vec_env/base_vec_env.py -------------------------------------------------------------------------------- /rl/utils/vec_env/dummy_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/vec_env/dummy_vec_env.py -------------------------------------------------------------------------------- /rl/utils/vec_env/subproc_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/vec_env/subproc_vec_env.py -------------------------------------------------------------------------------- /rl/utils/vec_env/vec_frame_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/vec_env/vec_frame_stack.py -------------------------------------------------------------------------------- /rl/utils/vec_env/vec_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/rl/utils/vec_env/vec_normalize.py -------------------------------------------------------------------------------- /scripts/distractor.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/scripts/distractor.sh -------------------------------------------------------------------------------- /scripts/inside_box.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/scripts/inside_box.sh -------------------------------------------------------------------------------- /scripts/pick.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/scripts/pick.sh -------------------------------------------------------------------------------- /scripts/train_antmaze.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/scripts/train_antmaze.sh -------------------------------------------------------------------------------- /scripts/train_pointmaze.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LunjunZhang/world-model-as-a-graph/HEAD/scripts/train_pointmaze.sh --------------------------------------------------------------------------------