Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Integrate bigym training pipeline #1

Merged
merged 12 commits into from
Aug 5, 2024
2 changes: 1 addition & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,4 +31,4 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0

### Fixed

- None.
- None.
11 changes: 8 additions & 3 deletions robobase/cfgs/env/bigym.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,13 @@

env:
env_name: bigym
episode_length: 1000
episode_length: 3000
cameras: ["head", "right_wrist", "left_wrist"]
action_mode: JOINT_POSITION
action_mode: absolute
floating: true
dataset_root: ""
demo_down_sample_rate: 20
render_mode: rgb_array
enable_all_floating_dof: false

action_repeat: 2
demos: !!float .inf
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/cupboards_close_all.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: cupboards_close_all
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 15500
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/cupboards_open_all.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: cupboards_open_all
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 22500
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_close.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_close
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 7500
demo_down_sample_rate: 20
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_close_trays.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_close_trays
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 3000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_load_cups.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_load_cups
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 7500
demo_down_sample_rate: 10
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_load_cutlery.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_load_cutlery
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 7000
demo_down_sample_rate: 10
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_load_plates.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_load_plates
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 14000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_open.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_open
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 7500
demo_down_sample_rate: 20
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_open_trays.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_open_trays
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 3000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_unload_cups.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_unload_cups
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 10000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_unload_cups_long.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_unload_cups_long
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 18000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_unload_cutlery.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_unload_cutlery
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 15500
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_unload_cutlery_long.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_unload_cutlery_long
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 18000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_unload_plates.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_unload_plates
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 20000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/dishwasher_unload_plates_long.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: dishwasher_unload_plates_long
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 26000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/drawer_top_close.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: drawer_top_close
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 3000
demo_down_sample_rate: 10
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/drawer_top_open.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: drawer_top_open
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 5000
demo_down_sample_rate: 10
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/drawers_close_all.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: drawers_close_all
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 5000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/drawers_open_all.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: drawers_open_all
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 12000
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/flip_cup.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: flip_cup
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 5500
demo_down_sample_rate: 10
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/flip_cutlery.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: flip_cutlery
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 12500
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/move_plate.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: move_plate
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 3000
demo_down_sample_rate: 10
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/move_two_plates.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: move_two_plates
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 5500
demo_down_sample_rate: 10
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/pick_box.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: pick_box
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 13500
demo_down_sample_rate: 25
enable_all_floating_dof: true
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/put_cups.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: put_cups
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 8500
demo_down_sample_rate: 20
enable_all_floating_dof: true
9 changes: 0 additions & 9 deletions robobase/cfgs/env/bigym/put_plate_in_drainer.yaml

This file was deleted.

9 changes: 0 additions & 9 deletions robobase/cfgs/env/bigym/reach_target.yaml

This file was deleted.

12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/reach_target_dual.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: reach_target_dual
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 3000
demo_down_sample_rate: 10
enable_all_floating_dof: false
12 changes: 12 additions & 0 deletions robobase/cfgs/env/bigym/reach_target_multi_modal.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
# @package _global_

defaults:
- bigym
- _self_

env:
task_name: reach_target_multi_modal
stddev_schedule: linear(1.0,0.1,500000)
episode_length: 3000
demo_down_sample_rate: 25
enable_all_floating_dof: false
Loading