├── .github └── workflows │ ├── robobase.yml │ └── robobase_integration.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CHANGELOG.md ├── README.md ├── extra_install_scripts └── install_coppeliasim.sh ├── robobase ├── __init__.py ├── cfgs │ ├── __init__.py │ ├── env │ │ ├── bigym.yaml │ │ ├── bigym │ │ │ ├── cupboards_close_all.yaml │ │ │ ├── cupboards_open_all.yaml │ │ │ ├── dishwasher_close.yaml │ │ │ ├── dishwasher_close_trays.yaml │ │ │ ├── dishwasher_load_cups.yaml │ │ │ ├── dishwasher_load_cutlery.yaml │ │ │ ├── dishwasher_load_plates.yaml │ │ │ ├── dishwasher_open.yaml │ │ │ ├── dishwasher_open_trays.yaml │ │ │ ├── dishwasher_unload_cups.yaml │ │ │ ├── dishwasher_unload_cups_long.yaml │ │ │ ├── dishwasher_unload_cutlery.yaml │ │ │ ├── dishwasher_unload_cutlery_long.yaml │ │ │ ├── dishwasher_unload_plates.yaml │ │ │ ├── dishwasher_unload_plates_long.yaml │ │ │ ├── drawer_top_close.yaml │ │ │ ├── drawer_top_open.yaml │ │ │ ├── drawers_close_all.yaml │ │ │ ├── drawers_open_all.yaml │ │ │ ├── flip_cup.yaml │ │ │ ├── flip_cutlery.yaml │ │ │ ├── move_plate.yaml │ │ │ ├── move_two_plates.yaml │ │ │ ├── pick_box.yaml │ │ │ ├── put_cups.yaml │ │ │ ├── reach_target_dual.yaml │ │ │ ├── reach_target_multi_modal.yaml │ │ │ ├── reach_target_single.yaml │ │ │ ├── sandwich_flip.yaml │ │ │ ├── sandwich_remove.yaml │ │ │ ├── sandwich_toast.yaml │ │ │ ├── saucepan_to_hob.yaml │ │ │ ├── stack_blocks.yaml │ │ │ ├── store_box.yaml │ │ │ ├── store_groceries_lower.yaml │ │ │ ├── store_groceries_upper.yaml │ │ │ ├── store_kitchenware.yaml │ │ │ ├── take_cups.yaml │ │ │ ├── wall_cupboard_close.yaml │ │ │ └── wall_cupboard_open.yaml │ │ ├── d4rl.yaml │ │ ├── d4rl │ │ │ ├── ant.yaml │ │ │ ├── antmaze.yaml │ │ │ ├── halfcheetah.yaml │ │ │ ├── hopper.yaml │ │ │ └── walker2d.yaml │ │ ├── dmc.yaml │ │ ├── dmc │ │ │ ├── acrobot_swingup.yaml │ │ │ ├── cartpole_balance.yaml │ │ │ ├── cartpole_balance_sparse.yaml │ │ │ ├── cartpole_swingup.yaml │ │ │ ├── cartpole_swingup_sparse.yaml │ │ │ ├── cheetah_run.yaml │ │ │ ├── cup_catch.yaml │ │ │ ├── easy.yaml │ │ │ ├── finger_spin.yaml │ │ │ ├── finger_turn_easy.yaml │ │ │ ├── finger_turn_hard.yaml │ │ │ ├── hard.yaml │ │ │ ├── hopper_hop.yaml │ │ │ ├── hopper_stand.yaml │ │ │ ├── humanoid_run.yaml │ │ │ ├── humanoid_stand.yaml │ │ │ ├── humanoid_walk.yaml │ │ │ ├── lift_brick.yaml │ │ │ ├── medium.yaml │ │ │ ├── pendulum_swingup.yaml │ │ │ ├── quadruped_run.yaml │ │ │ ├── quadruped_walk.yaml │ │ │ ├── reach_duplo.yaml │ │ │ ├── reacher_easy.yaml │ │ │ ├── reacher_hard.yaml │ │ │ ├── stacker_stack_2.yaml │ │ │ ├── walker_run.yaml │ │ │ ├── walker_stand.yaml │ │ │ └── walker_walk.yaml │ │ ├── rlbench.yaml │ │ └── rlbench │ │ │ ├── basketball_in_hoop.yaml │ │ │ ├── easy.yaml │ │ │ ├── episode_length │ │ │ ├── cqn.yaml │ │ │ └── default.yaml │ │ │ ├── insert_usb_in_computer.yaml │ │ │ ├── lamp_on.yaml │ │ │ ├── meat_on_grill.yaml │ │ │ ├── open_door.yaml │ │ │ ├── open_drawer.yaml │ │ │ ├── open_microwave.yaml │ │ │ ├── open_oven.yaml │ │ │ ├── phone_on_base.yaml │ │ │ ├── pick_up_cup.yaml │ │ │ ├── press_switch.yaml │ │ │ ├── put_books_on_bookshelf.yaml │ │ │ ├── put_item_in_drawer.yaml │ │ │ ├── put_money_in_safe.yaml │ │ │ ├── put_rubbish_in_bin.yaml │ │ │ ├── reach_target.yaml │ │ │ ├── slide_block_to_target.yaml │ │ │ ├── stack_wine.yaml │ │ │ ├── sweep_to_dustpan.yaml │ │ │ ├── take_lid_off_saucepan.yaml │ │ │ ├── take_plate_off_colored_dish_rack.yaml │ │ │ ├── toilet_seat_up.yaml │ │ │ └── turn_tap.yaml │ ├── intrinsic_reward_module │ │ ├── icm.yaml │ │ └── rnd.yaml │ ├── launch │ │ ├── act_pixel_bigym.yaml │ │ ├── act_pixel_rlbench.yaml │ │ ├── alix_pixel_dmc.yaml │ │ ├── cqn_pixel_dmc.yaml │ │ ├── cqn_pixel_rlbench_demo_driven.yaml │ │ ├── cqn_state_dmc.yaml │ │ ├── cqn_state_rlbench_demo_driven.yaml │ │ ├── dp_pixel_bigym.yaml │ │ ├── dreamerv3_pixel_dmc.yaml │ │ ├── drm_pixel_dmc.yaml │ │ ├── drqv2.yaml │ │ ├── drqv2_dist_critic.yaml │ │ ├── drqv2_pixel_dmc.yaml │ │ ├── drqv2plus_pixel_rlbench_demo_driven.yaml │ │ ├── edp_d4rl.yaml │ │ ├── iql_drqv2.yaml │ │ ├── mwm.yaml │ │ ├── mwm_pixel_dmc.yaml │ │ └── sac_lix_pixel_dmc.yaml │ ├── method │ │ ├── act.yaml │ │ ├── alix.yaml │ │ ├── bc.yaml │ │ ├── cqn.yaml │ │ ├── diffusion.yaml │ │ ├── dreamerv3.yaml │ │ ├── drm.yaml │ │ ├── drqv2.yaml │ │ ├── edp.yaml │ │ ├── iql_drqv2.yaml │ │ ├── mwm.yaml │ │ ├── sac_lix.yaml │ │ └── value_based.yaml │ └── robobase_config.yaml ├── envs │ ├── __init__.py │ ├── bigym.py │ ├── d4rl.py │ ├── dmc.py │ ├── env.py │ ├── rlbench.py │ ├── utils │ │ └── bigym_utils.py │ └── wrappers │ │ ├── __init__.py │ │ ├── action_sequence.py │ │ ├── append_demo_info.py │ │ ├── concat_dim.py │ │ ├── frame_stack.py │ │ ├── onehot_time.py │ │ ├── rescale_from_tanh.py │ │ ├── reward_modifiers.py │ │ └── transpose_image_chw.py ├── intrinsic_reward_module │ ├── __init__.py │ ├── core.py │ ├── icm.py │ ├── rnd.py │ └── utils.py ├── logger.py ├── method │ ├── __init__.py │ ├── act.py │ ├── actor_critic.py │ ├── alix.py │ ├── bc.py │ ├── core.py │ ├── cqn.py │ ├── diffusion.py │ ├── dreamerv3.py │ ├── drm.py │ ├── drqv2.py │ ├── edp.py │ ├── iql_drqv2.py │ ├── mwm.py │ ├── sac_lix.py │ ├── utils.py │ └── value_based.py ├── models │ ├── __init__.py │ ├── act │ │ ├── __init__.py │ │ ├── backbone.py │ │ ├── position_encoding.py │ │ ├── transformer.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── misc.py │ │ │ └── resnet_film.py │ ├── core.py │ ├── decoder.py │ ├── diffusion_models.py │ ├── encoder.py │ ├── fully_connected.py │ ├── fusion.py │ ├── lix_utils │ │ ├── __init__.py │ │ ├── analysis_custom_autograd_functions.py │ │ ├── analysis_layers.py │ │ ├── analysis_modules.py │ │ ├── analysis_optimizers.py │ │ └── analysis_utils.py │ ├── model_based │ │ ├── distributions.py │ │ ├── dynamics_model.py │ │ └── utils.py │ ├── multi_view_transformer.py │ └── utils.py ├── replay_buffer │ ├── __init__.py │ ├── prioritized_replay_buffer.py │ ├── replay_buffer.py │ ├── sum_tree.py │ ├── uniform_replay_buffer.py │ └── utils.py ├── utils.py ├── video.py └── workspace.py ├── setup.py ├── tests ├── __init__.py ├── conftest.py ├── integration │ └── test_training.py └── unit │ ├── __init__.py │ ├── envs │ ├── test_d4rl.py │ └── test_rlbench.py │ ├── intrinsic_reward_module │ └── test_intrinsic_rewards.py │ ├── method │ ├── base.py │ ├── test_il_methods.py │ ├── test_model_based_rl_methods.py │ └── test_rl_methods.py │ ├── models │ └── test_models.py │ ├── replay_buffer │ ├── test_prioritized_replay_buffer.py │ ├── test_sequential_replay_buffer.py │ ├── test_sum_tree.py │ └── test_uniform_replay_buffer.py │ ├── test_workspace.py │ ├── utils │ └── test_add_demo_to_replay_buffer.py │ └── wrappers │ ├── __init__.py │ ├── test_action_sequence.py │ ├── test_concat_dim.py │ ├── test_frame_stack.py │ ├── test_onehot_time.py │ ├── test_rescale_from_tanh.py │ ├── test_reward_modifiers.py │ └── utils.py └── train.py /.github/workflows/robobase.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/.github/workflows/robobase.yml -------------------------------------------------------------------------------- /.github/workflows/robobase_integration.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/.github/workflows/robobase_integration.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/README.md -------------------------------------------------------------------------------- /extra_install_scripts/install_coppeliasim.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/extra_install_scripts/install_coppeliasim.sh -------------------------------------------------------------------------------- /robobase/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = "1.0.0" 2 | -------------------------------------------------------------------------------- /robobase/cfgs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/__init__.py -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/cupboards_close_all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/cupboards_close_all.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/cupboards_open_all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/cupboards_open_all.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_close.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_close.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_close_trays.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_close_trays.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_load_cups.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_load_cups.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_load_cutlery.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_load_cutlery.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_load_plates.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_load_plates.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_open.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_open.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_open_trays.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_open_trays.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_unload_cups.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_unload_cups.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_unload_cups_long.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_unload_cups_long.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_unload_cutlery.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_unload_cutlery.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_unload_cutlery_long.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_unload_cutlery_long.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_unload_plates.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_unload_plates.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/dishwasher_unload_plates_long.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/dishwasher_unload_plates_long.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/drawer_top_close.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/drawer_top_close.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/drawer_top_open.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/drawer_top_open.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/drawers_close_all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/drawers_close_all.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/drawers_open_all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/drawers_open_all.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/flip_cup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/flip_cup.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/flip_cutlery.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/flip_cutlery.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/move_plate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/move_plate.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/move_two_plates.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/move_two_plates.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/pick_box.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/pick_box.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/put_cups.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/put_cups.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/reach_target_dual.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/reach_target_dual.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/reach_target_multi_modal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/reach_target_multi_modal.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/reach_target_single.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/reach_target_single.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/sandwich_flip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/sandwich_flip.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/sandwich_remove.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/sandwich_remove.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/sandwich_toast.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/sandwich_toast.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/saucepan_to_hob.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/saucepan_to_hob.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/stack_blocks.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/stack_blocks.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/store_box.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/store_box.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/store_groceries_lower.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/store_groceries_lower.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/store_groceries_upper.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/store_groceries_upper.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/store_kitchenware.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/store_kitchenware.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/take_cups.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/take_cups.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/wall_cupboard_close.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/wall_cupboard_close.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/bigym/wall_cupboard_open.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/bigym/wall_cupboard_open.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/d4rl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/d4rl.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/d4rl/ant.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/d4rl/ant.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/d4rl/antmaze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/d4rl/antmaze.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/d4rl/halfcheetah.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/d4rl/halfcheetah.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/d4rl/hopper.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/d4rl/hopper.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/d4rl/walker2d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/d4rl/walker2d.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/acrobot_swingup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/acrobot_swingup.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/cartpole_balance.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/cartpole_balance.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/cartpole_balance_sparse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/cartpole_balance_sparse.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/cartpole_swingup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/cartpole_swingup.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/cartpole_swingup_sparse.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/cartpole_swingup_sparse.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/cheetah_run.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/cheetah_run.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/cup_catch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/cup_catch.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/easy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/easy.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/finger_spin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/finger_spin.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/finger_turn_easy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/finger_turn_easy.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/finger_turn_hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/finger_turn_hard.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/hard.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/hopper_hop.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/hopper_hop.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/hopper_stand.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/hopper_stand.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/humanoid_run.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/humanoid_run.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/humanoid_stand.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/humanoid_stand.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/humanoid_walk.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/humanoid_walk.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/lift_brick.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/lift_brick.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/medium.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/pendulum_swingup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/pendulum_swingup.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/quadruped_run.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/quadruped_run.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/quadruped_walk.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/quadruped_walk.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/reach_duplo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/reach_duplo.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/reacher_easy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/reacher_easy.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/reacher_hard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/reacher_hard.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/stacker_stack_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/stacker_stack_2.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/walker_run.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/walker_run.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/walker_stand.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/walker_stand.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/dmc/walker_walk.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/dmc/walker_walk.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/basketball_in_hoop.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/basketball_in_hoop.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/easy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/easy.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/episode_length/cqn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/episode_length/cqn.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/episode_length/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/episode_length/default.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/insert_usb_in_computer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/insert_usb_in_computer.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/lamp_on.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/lamp_on.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/meat_on_grill.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/meat_on_grill.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/open_door.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/open_door.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/open_drawer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/open_drawer.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/open_microwave.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/open_microwave.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/open_oven.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/open_oven.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/phone_on_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/phone_on_base.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/pick_up_cup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/pick_up_cup.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/press_switch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/press_switch.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/put_books_on_bookshelf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/put_books_on_bookshelf.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/put_item_in_drawer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/put_item_in_drawer.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/put_money_in_safe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/put_money_in_safe.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/put_rubbish_in_bin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/put_rubbish_in_bin.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/reach_target.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/reach_target.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/slide_block_to_target.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/slide_block_to_target.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/stack_wine.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/stack_wine.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/sweep_to_dustpan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/sweep_to_dustpan.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/take_lid_off_saucepan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/take_lid_off_saucepan.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/take_plate_off_colored_dish_rack.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/take_plate_off_colored_dish_rack.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/toilet_seat_up.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/toilet_seat_up.yaml -------------------------------------------------------------------------------- /robobase/cfgs/env/rlbench/turn_tap.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/env/rlbench/turn_tap.yaml -------------------------------------------------------------------------------- /robobase/cfgs/intrinsic_reward_module/icm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/intrinsic_reward_module/icm.yaml -------------------------------------------------------------------------------- /robobase/cfgs/intrinsic_reward_module/rnd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/intrinsic_reward_module/rnd.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/act_pixel_bigym.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/act_pixel_bigym.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/act_pixel_rlbench.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/act_pixel_rlbench.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/alix_pixel_dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/alix_pixel_dmc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/cqn_pixel_dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/cqn_pixel_dmc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/cqn_pixel_rlbench_demo_driven.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/cqn_pixel_rlbench_demo_driven.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/cqn_state_dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/cqn_state_dmc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/cqn_state_rlbench_demo_driven.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/cqn_state_rlbench_demo_driven.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/dp_pixel_bigym.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/dp_pixel_bigym.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/dreamerv3_pixel_dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/dreamerv3_pixel_dmc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/drm_pixel_dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/drm_pixel_dmc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/drqv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/drqv2.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/drqv2_dist_critic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/drqv2_dist_critic.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/drqv2_pixel_dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/drqv2_pixel_dmc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/drqv2plus_pixel_rlbench_demo_driven.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/drqv2plus_pixel_rlbench_demo_driven.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/edp_d4rl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/edp_d4rl.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/iql_drqv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/iql_drqv2.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/mwm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/mwm.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/mwm_pixel_dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/mwm_pixel_dmc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/launch/sac_lix_pixel_dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/launch/sac_lix_pixel_dmc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/act.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/act.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/alix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/alix.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/bc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/bc.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/cqn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/cqn.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/diffusion.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/diffusion.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/dreamerv3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/dreamerv3.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/drm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/drm.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/drqv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/drqv2.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/edp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/edp.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/iql_drqv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/iql_drqv2.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/mwm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/mwm.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/sac_lix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/sac_lix.yaml -------------------------------------------------------------------------------- /robobase/cfgs/method/value_based.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/method/value_based.yaml -------------------------------------------------------------------------------- /robobase/cfgs/robobase_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/cfgs/robobase_config.yaml -------------------------------------------------------------------------------- /robobase/envs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /robobase/envs/bigym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/bigym.py -------------------------------------------------------------------------------- /robobase/envs/d4rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/d4rl.py -------------------------------------------------------------------------------- /robobase/envs/dmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/dmc.py -------------------------------------------------------------------------------- /robobase/envs/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/env.py -------------------------------------------------------------------------------- /robobase/envs/rlbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/rlbench.py -------------------------------------------------------------------------------- /robobase/envs/utils/bigym_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/utils/bigym_utils.py -------------------------------------------------------------------------------- /robobase/envs/wrappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/wrappers/__init__.py -------------------------------------------------------------------------------- /robobase/envs/wrappers/action_sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/wrappers/action_sequence.py -------------------------------------------------------------------------------- /robobase/envs/wrappers/append_demo_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/wrappers/append_demo_info.py -------------------------------------------------------------------------------- /robobase/envs/wrappers/concat_dim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/wrappers/concat_dim.py -------------------------------------------------------------------------------- /robobase/envs/wrappers/frame_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/wrappers/frame_stack.py -------------------------------------------------------------------------------- /robobase/envs/wrappers/onehot_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/wrappers/onehot_time.py -------------------------------------------------------------------------------- /robobase/envs/wrappers/rescale_from_tanh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/wrappers/rescale_from_tanh.py -------------------------------------------------------------------------------- /robobase/envs/wrappers/reward_modifiers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/wrappers/reward_modifiers.py -------------------------------------------------------------------------------- /robobase/envs/wrappers/transpose_image_chw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/envs/wrappers/transpose_image_chw.py -------------------------------------------------------------------------------- /robobase/intrinsic_reward_module/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/intrinsic_reward_module/__init__.py -------------------------------------------------------------------------------- /robobase/intrinsic_reward_module/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/intrinsic_reward_module/core.py -------------------------------------------------------------------------------- /robobase/intrinsic_reward_module/icm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/intrinsic_reward_module/icm.py -------------------------------------------------------------------------------- /robobase/intrinsic_reward_module/rnd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/intrinsic_reward_module/rnd.py -------------------------------------------------------------------------------- /robobase/intrinsic_reward_module/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/intrinsic_reward_module/utils.py -------------------------------------------------------------------------------- /robobase/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/logger.py -------------------------------------------------------------------------------- /robobase/method/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /robobase/method/act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/act.py -------------------------------------------------------------------------------- /robobase/method/actor_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/actor_critic.py -------------------------------------------------------------------------------- /robobase/method/alix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/alix.py -------------------------------------------------------------------------------- /robobase/method/bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/bc.py -------------------------------------------------------------------------------- /robobase/method/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/core.py -------------------------------------------------------------------------------- /robobase/method/cqn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/cqn.py -------------------------------------------------------------------------------- /robobase/method/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/diffusion.py -------------------------------------------------------------------------------- /robobase/method/dreamerv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/dreamerv3.py -------------------------------------------------------------------------------- /robobase/method/drm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/drm.py -------------------------------------------------------------------------------- /robobase/method/drqv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/drqv2.py -------------------------------------------------------------------------------- /robobase/method/edp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/edp.py -------------------------------------------------------------------------------- /robobase/method/iql_drqv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/iql_drqv2.py -------------------------------------------------------------------------------- /robobase/method/mwm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/mwm.py -------------------------------------------------------------------------------- /robobase/method/sac_lix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/sac_lix.py -------------------------------------------------------------------------------- /robobase/method/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/utils.py -------------------------------------------------------------------------------- /robobase/method/value_based.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/method/value_based.py -------------------------------------------------------------------------------- /robobase/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/__init__.py -------------------------------------------------------------------------------- /robobase/models/act/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /robobase/models/act/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/act/backbone.py -------------------------------------------------------------------------------- /robobase/models/act/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/act/position_encoding.py -------------------------------------------------------------------------------- /robobase/models/act/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/act/transformer.py -------------------------------------------------------------------------------- /robobase/models/act/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /robobase/models/act/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/act/utils/misc.py -------------------------------------------------------------------------------- /robobase/models/act/utils/resnet_film.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/act/utils/resnet_film.py -------------------------------------------------------------------------------- /robobase/models/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/core.py -------------------------------------------------------------------------------- /robobase/models/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/decoder.py -------------------------------------------------------------------------------- /robobase/models/diffusion_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/diffusion_models.py -------------------------------------------------------------------------------- /robobase/models/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/encoder.py -------------------------------------------------------------------------------- /robobase/models/fully_connected.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/fully_connected.py -------------------------------------------------------------------------------- /robobase/models/fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/fusion.py -------------------------------------------------------------------------------- /robobase/models/lix_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /robobase/models/lix_utils/analysis_custom_autograd_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/lix_utils/analysis_custom_autograd_functions.py -------------------------------------------------------------------------------- /robobase/models/lix_utils/analysis_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/lix_utils/analysis_layers.py -------------------------------------------------------------------------------- /robobase/models/lix_utils/analysis_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/lix_utils/analysis_modules.py -------------------------------------------------------------------------------- /robobase/models/lix_utils/analysis_optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/lix_utils/analysis_optimizers.py -------------------------------------------------------------------------------- /robobase/models/lix_utils/analysis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/lix_utils/analysis_utils.py -------------------------------------------------------------------------------- /robobase/models/model_based/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/model_based/distributions.py -------------------------------------------------------------------------------- /robobase/models/model_based/dynamics_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/model_based/dynamics_model.py -------------------------------------------------------------------------------- /robobase/models/model_based/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/model_based/utils.py -------------------------------------------------------------------------------- /robobase/models/multi_view_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/multi_view_transformer.py -------------------------------------------------------------------------------- /robobase/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/models/utils.py -------------------------------------------------------------------------------- /robobase/replay_buffer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /robobase/replay_buffer/prioritized_replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/replay_buffer/prioritized_replay_buffer.py -------------------------------------------------------------------------------- /robobase/replay_buffer/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/replay_buffer/replay_buffer.py -------------------------------------------------------------------------------- /robobase/replay_buffer/sum_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/replay_buffer/sum_tree.py -------------------------------------------------------------------------------- /robobase/replay_buffer/uniform_replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/replay_buffer/uniform_replay_buffer.py -------------------------------------------------------------------------------- /robobase/replay_buffer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/replay_buffer/utils.py -------------------------------------------------------------------------------- /robobase/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/utils.py -------------------------------------------------------------------------------- /robobase/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/video.py -------------------------------------------------------------------------------- /robobase/workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/robobase/workspace.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/integration/test_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/integration/test_training.py -------------------------------------------------------------------------------- /tests/unit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unit/envs/test_d4rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/envs/test_d4rl.py -------------------------------------------------------------------------------- /tests/unit/envs/test_rlbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/envs/test_rlbench.py -------------------------------------------------------------------------------- /tests/unit/intrinsic_reward_module/test_intrinsic_rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/intrinsic_reward_module/test_intrinsic_rewards.py -------------------------------------------------------------------------------- /tests/unit/method/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/method/base.py -------------------------------------------------------------------------------- /tests/unit/method/test_il_methods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/method/test_il_methods.py -------------------------------------------------------------------------------- /tests/unit/method/test_model_based_rl_methods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/method/test_model_based_rl_methods.py -------------------------------------------------------------------------------- /tests/unit/method/test_rl_methods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/method/test_rl_methods.py -------------------------------------------------------------------------------- /tests/unit/models/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/models/test_models.py -------------------------------------------------------------------------------- /tests/unit/replay_buffer/test_prioritized_replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/replay_buffer/test_prioritized_replay_buffer.py -------------------------------------------------------------------------------- /tests/unit/replay_buffer/test_sequential_replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/replay_buffer/test_sequential_replay_buffer.py -------------------------------------------------------------------------------- /tests/unit/replay_buffer/test_sum_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/replay_buffer/test_sum_tree.py -------------------------------------------------------------------------------- /tests/unit/replay_buffer/test_uniform_replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/replay_buffer/test_uniform_replay_buffer.py -------------------------------------------------------------------------------- /tests/unit/test_workspace.py: -------------------------------------------------------------------------------- 1 | # TODO: Test if workspace does pre-training steps, etc 2 | -------------------------------------------------------------------------------- /tests/unit/utils/test_add_demo_to_replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/utils/test_add_demo_to_replay_buffer.py -------------------------------------------------------------------------------- /tests/unit/wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unit/wrappers/test_action_sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/wrappers/test_action_sequence.py -------------------------------------------------------------------------------- /tests/unit/wrappers/test_concat_dim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/wrappers/test_concat_dim.py -------------------------------------------------------------------------------- /tests/unit/wrappers/test_frame_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/wrappers/test_frame_stack.py -------------------------------------------------------------------------------- /tests/unit/wrappers/test_onehot_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/wrappers/test_onehot_time.py -------------------------------------------------------------------------------- /tests/unit/wrappers/test_rescale_from_tanh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/wrappers/test_rescale_from_tanh.py -------------------------------------------------------------------------------- /tests/unit/wrappers/test_reward_modifiers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/wrappers/test_reward_modifiers.py -------------------------------------------------------------------------------- /tests/unit/wrappers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/tests/unit/wrappers/utils.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robobase-org/robobase/HEAD/train.py --------------------------------------------------------------------------------