Upload 20 files
Browse files- dp/prepare_groceries/pick/all/config.yml +63 -0
- dp/prepare_groceries/pick/all/policy.pt +3 -0
- dp/prepare_groceries/place/all/config.yml +63 -0
- dp/prepare_groceries/place/all/policy.pt +3 -0
- dp/set_table/close/fridge/config.yml +64 -0
- dp/set_table/close/fridge/policy.pt +3 -0
- dp/set_table/close/kitchen_counter/config.yml +64 -0
- dp/set_table/close/kitchen_counter/policy.pt +3 -0
- dp/set_table/open/fridge/config.yml +64 -0
- dp/set_table/open/fridge/policy.pt +3 -0
- dp/set_table/open/kitchen_counter/config.yml +64 -0
- dp/set_table/open/kitchen_counter/policy.pt +3 -0
- dp/set_table/pick/all/config.yml +63 -0
- dp/set_table/pick/all/policy.pt +3 -0
- dp/set_table/place/all/config.yml +63 -0
- dp/set_table/place/all/policy.pt +3 -0
- dp/tidy_house/pick/all/config.yml +63 -0
- dp/tidy_house/pick/all/policy.pt +3 -0
- dp/tidy_house/place/all/config.yml +63 -0
- dp/tidy_house/place/all/policy.pt +3 -0
dp/prepare_groceries/pick/all/config.yml
ADDED
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: PickSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/prepare_groceries/pick/train/all.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/prepare_groceries/pick/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- ee_rest
|
25 |
+
- robot_rest
|
26 |
+
- is_static
|
27 |
+
- cumulative_force_within_limit
|
28 |
+
env_kwargs:
|
29 |
+
robot_force_mult: 0.001
|
30 |
+
robot_force_penalty_min: 0.2
|
31 |
+
target_randomization: false
|
32 |
+
algo:
|
33 |
+
name: diffusion_policy
|
34 |
+
lr: 0.0001
|
35 |
+
batch_size: 256
|
36 |
+
obs_horizon: 2
|
37 |
+
act_horizon: 1
|
38 |
+
pred_horizon: 16
|
39 |
+
diffusion_step_embed_dim: 256
|
40 |
+
unet_dims:
|
41 |
+
- 256
|
42 |
+
- 512
|
43 |
+
- 1024
|
44 |
+
n_groups: 8
|
45 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/prepare_groceries/pick
|
46 |
+
trajs_per_obj: all
|
47 |
+
truncate_trajectories_at_success: true
|
48 |
+
max_image_cache_size: all
|
49 |
+
num_dataload_workers: 2
|
50 |
+
num_iterations: 500000
|
51 |
+
eval_episodes: 63
|
52 |
+
log_freq: 1000
|
53 |
+
eval_freq: 5000
|
54 |
+
save_freq: 5000
|
55 |
+
torch_deterministic: true
|
56 |
+
save_backup_ckpts: false
|
57 |
+
logger:
|
58 |
+
workspace: mshab_exps
|
59 |
+
exp_name: rcad-prepare_groceries-pick-all
|
60 |
+
clear_out: true
|
61 |
+
tensorboard: true
|
62 |
+
wandb: false
|
63 |
+
model_ckpt: "mshab_checkpoints/dp/prepare_groceries/pick/all/policy.pt"
|
dp/prepare_groceries/pick/all/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7d3299c79843c4dcbee31e71276c3aaaa6f546e3d0cd1e2120f429bac1b001b9
|
3 |
+
size 1183023810
|
dp/prepare_groceries/place/all/config.yml
ADDED
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: PlaceSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/prepare_groceries/place/train/all.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/prepare_groceries/place/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- ee_rest
|
25 |
+
- robot_rest
|
26 |
+
- is_static
|
27 |
+
- cumulative_force_within_limit
|
28 |
+
env_kwargs:
|
29 |
+
robot_force_mult: 0.001
|
30 |
+
robot_force_penalty_min: 0.2
|
31 |
+
target_randomization: false
|
32 |
+
algo:
|
33 |
+
name: diffusion_policy
|
34 |
+
lr: 0.0001
|
35 |
+
batch_size: 256
|
36 |
+
obs_horizon: 2
|
37 |
+
act_horizon: 1
|
38 |
+
pred_horizon: 16
|
39 |
+
diffusion_step_embed_dim: 256
|
40 |
+
unet_dims:
|
41 |
+
- 256
|
42 |
+
- 512
|
43 |
+
- 1024
|
44 |
+
n_groups: 8
|
45 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/prepare_groceries/place
|
46 |
+
trajs_per_obj: all
|
47 |
+
truncate_trajectories_at_success: true
|
48 |
+
max_image_cache_size: all
|
49 |
+
num_dataload_workers: 2
|
50 |
+
num_iterations: 500000
|
51 |
+
eval_episodes: 63
|
52 |
+
log_freq: 1000
|
53 |
+
eval_freq: 5000
|
54 |
+
save_freq: 5000
|
55 |
+
torch_deterministic: true
|
56 |
+
save_backup_ckpts: false
|
57 |
+
logger:
|
58 |
+
workspace: mshab_exps
|
59 |
+
exp_name: rcad-prepare_groceries-place-all
|
60 |
+
clear_out: true
|
61 |
+
tensorboard: true
|
62 |
+
wandb: false
|
63 |
+
model_ckpt: "mshab_checkpoints/dp/prepare_groceries/place/all/policy.pt"
|
dp/prepare_groceries/place/all/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ae3421b5fe11451ff565634a2304a1021dbe8f07f638e3d4260d38e7ba45447d
|
3 |
+
size 1183023810
|
dp/set_table/close/fridge/config.yml
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: CloseSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/set_table/close/train/fridge.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/set_table/close/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- articulation_closed
|
25 |
+
- ee_rest
|
26 |
+
- robot_rest
|
27 |
+
- is_static
|
28 |
+
- cumulative_force_within_limit
|
29 |
+
env_kwargs:
|
30 |
+
robot_force_mult: 0.001
|
31 |
+
robot_force_penalty_min: 0.2
|
32 |
+
target_randomization: false
|
33 |
+
algo:
|
34 |
+
name: diffusion_policy
|
35 |
+
lr: 0.0001
|
36 |
+
batch_size: 256
|
37 |
+
obs_horizon: 2
|
38 |
+
act_horizon: 1
|
39 |
+
pred_horizon: 16
|
40 |
+
diffusion_step_embed_dim: 256
|
41 |
+
unet_dims:
|
42 |
+
- 256
|
43 |
+
- 512
|
44 |
+
- 1024
|
45 |
+
n_groups: 8
|
46 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/set_table/close/fridge.h5
|
47 |
+
trajs_per_obj: all
|
48 |
+
truncate_trajectories_at_success: true
|
49 |
+
max_image_cache_size: all
|
50 |
+
num_dataload_workers: 2
|
51 |
+
num_iterations: 500000
|
52 |
+
eval_episodes: 63
|
53 |
+
log_freq: 1000
|
54 |
+
eval_freq: 5000
|
55 |
+
save_freq: 5000
|
56 |
+
torch_deterministic: true
|
57 |
+
save_backup_ckpts: false
|
58 |
+
logger:
|
59 |
+
workspace: mshab_exps
|
60 |
+
exp_name: rcad-set_table-close-fridge
|
61 |
+
clear_out: true
|
62 |
+
tensorboard: true
|
63 |
+
wandb: false
|
64 |
+
model_ckpt: "mshab_checkpoints/dp/set_table/close/fridge/policy.pt"
|
dp/set_table/close/fridge/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d74949915c9fff29ad13167e71df3c6c4d86ca63b11f264f4a19f7841011631b
|
3 |
+
size 1183023810
|
dp/set_table/close/kitchen_counter/config.yml
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: CloseSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/set_table/close/train/kitchen_counter.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/set_table/close/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- articulation_closed
|
25 |
+
- ee_rest
|
26 |
+
- robot_rest
|
27 |
+
- is_static
|
28 |
+
- cumulative_force_within_limit
|
29 |
+
env_kwargs:
|
30 |
+
robot_force_mult: 0.001
|
31 |
+
robot_force_penalty_min: 0.2
|
32 |
+
target_randomization: false
|
33 |
+
algo:
|
34 |
+
name: diffusion_policy
|
35 |
+
lr: 0.0001
|
36 |
+
batch_size: 256
|
37 |
+
obs_horizon: 2
|
38 |
+
act_horizon: 1
|
39 |
+
pred_horizon: 16
|
40 |
+
diffusion_step_embed_dim: 256
|
41 |
+
unet_dims:
|
42 |
+
- 256
|
43 |
+
- 512
|
44 |
+
- 1024
|
45 |
+
n_groups: 8
|
46 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/set_table/close/kitchen_counter.h5
|
47 |
+
trajs_per_obj: all
|
48 |
+
truncate_trajectories_at_success: true
|
49 |
+
max_image_cache_size: all
|
50 |
+
num_dataload_workers: 2
|
51 |
+
num_iterations: 500000
|
52 |
+
eval_episodes: 63
|
53 |
+
log_freq: 1000
|
54 |
+
eval_freq: 5000
|
55 |
+
save_freq: 5000
|
56 |
+
torch_deterministic: true
|
57 |
+
save_backup_ckpts: false
|
58 |
+
logger:
|
59 |
+
workspace: mshab_exps
|
60 |
+
exp_name: rcad-set_table-close-kitchen_counter
|
61 |
+
clear_out: true
|
62 |
+
tensorboard: true
|
63 |
+
wandb: false
|
64 |
+
model_ckpt: "mshab_checkpoints/dp/set_table/close/kitchen_counter/policy.pt"
|
dp/set_table/close/kitchen_counter/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c88f9552fb0f0dc99cf101eef3a5fcc094652f87ae764d4ac281c103c1213f92
|
3 |
+
size 1183023810
|
dp/set_table/open/fridge/config.yml
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: OpenSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/set_table/open/train/fridge.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/set_table/open/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- articulation_open
|
25 |
+
- ee_rest
|
26 |
+
- robot_rest
|
27 |
+
- is_static
|
28 |
+
- cumulative_force_within_limit
|
29 |
+
env_kwargs:
|
30 |
+
robot_force_mult: 0.001
|
31 |
+
robot_force_penalty_min: 0.2
|
32 |
+
target_randomization: false
|
33 |
+
algo:
|
34 |
+
name: diffusion_policy
|
35 |
+
lr: 0.0001
|
36 |
+
batch_size: 256
|
37 |
+
obs_horizon: 2
|
38 |
+
act_horizon: 1
|
39 |
+
pred_horizon: 16
|
40 |
+
diffusion_step_embed_dim: 256
|
41 |
+
unet_dims:
|
42 |
+
- 256
|
43 |
+
- 512
|
44 |
+
- 1024
|
45 |
+
n_groups: 8
|
46 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/set_table/open/fridge.h5
|
47 |
+
trajs_per_obj: all
|
48 |
+
truncate_trajectories_at_success: true
|
49 |
+
max_image_cache_size: all
|
50 |
+
num_dataload_workers: 2
|
51 |
+
num_iterations: 500000
|
52 |
+
eval_episodes: 63
|
53 |
+
log_freq: 1000
|
54 |
+
eval_freq: 5000
|
55 |
+
save_freq: 5000
|
56 |
+
torch_deterministic: true
|
57 |
+
save_backup_ckpts: false
|
58 |
+
logger:
|
59 |
+
workspace: mshab_exps
|
60 |
+
exp_name: rcad-set_table-open-fridge
|
61 |
+
clear_out: true
|
62 |
+
tensorboard: true
|
63 |
+
wandb: false
|
64 |
+
model_ckpt: "mshab_checkpoints/dp/set_table/open/fridge/policy.pt"
|
dp/set_table/open/fridge/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:63ee243cfdc11acad9eb1b2c668418121cf59e75d1feecd8f24f133818c892b1
|
3 |
+
size 1183023810
|
dp/set_table/open/kitchen_counter/config.yml
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: OpenSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/set_table/open/train/kitchen_counter.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/set_table/open/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- articulation_open
|
25 |
+
- ee_rest
|
26 |
+
- robot_rest
|
27 |
+
- is_static
|
28 |
+
- cumulative_force_within_limit
|
29 |
+
env_kwargs:
|
30 |
+
robot_force_mult: 0.001
|
31 |
+
robot_force_penalty_min: 0.2
|
32 |
+
target_randomization: false
|
33 |
+
algo:
|
34 |
+
name: diffusion_policy
|
35 |
+
lr: 0.0001
|
36 |
+
batch_size: 256
|
37 |
+
obs_horizon: 2
|
38 |
+
act_horizon: 1
|
39 |
+
pred_horizon: 16
|
40 |
+
diffusion_step_embed_dim: 256
|
41 |
+
unet_dims:
|
42 |
+
- 256
|
43 |
+
- 512
|
44 |
+
- 1024
|
45 |
+
n_groups: 8
|
46 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/set_table/open/kitchen_counter.h5
|
47 |
+
trajs_per_obj: all
|
48 |
+
truncate_trajectories_at_success: true
|
49 |
+
max_image_cache_size: all
|
50 |
+
num_dataload_workers: 2
|
51 |
+
num_iterations: 500000
|
52 |
+
eval_episodes: 63
|
53 |
+
log_freq: 1000
|
54 |
+
eval_freq: 5000
|
55 |
+
save_freq: 5000
|
56 |
+
torch_deterministic: true
|
57 |
+
save_backup_ckpts: false
|
58 |
+
logger:
|
59 |
+
workspace: mshab_exps
|
60 |
+
exp_name: rcad-set_table-open-kitchen_counter
|
61 |
+
clear_out: true
|
62 |
+
tensorboard: true
|
63 |
+
wandb: false
|
64 |
+
model_ckpt: "mshab_checkpoints/dp/set_table/open/kitchen_counter/policy.pt"
|
dp/set_table/open/kitchen_counter/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:de55c58b3a86efeea8aaf8cb93aca27b42a31633edf5885779cd9dcdbdb458e5
|
3 |
+
size 1183023810
|
dp/set_table/pick/all/config.yml
ADDED
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: PickSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/set_table/pick/train/all.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/set_table/pick/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- ee_rest
|
25 |
+
- robot_rest
|
26 |
+
- is_static
|
27 |
+
- cumulative_force_within_limit
|
28 |
+
env_kwargs:
|
29 |
+
robot_force_mult: 0.001
|
30 |
+
robot_force_penalty_min: 0.2
|
31 |
+
target_randomization: false
|
32 |
+
algo:
|
33 |
+
name: diffusion_policy
|
34 |
+
lr: 0.0001
|
35 |
+
batch_size: 256
|
36 |
+
obs_horizon: 2
|
37 |
+
act_horizon: 1
|
38 |
+
pred_horizon: 16
|
39 |
+
diffusion_step_embed_dim: 256
|
40 |
+
unet_dims:
|
41 |
+
- 256
|
42 |
+
- 512
|
43 |
+
- 1024
|
44 |
+
n_groups: 8
|
45 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/set_table/pick
|
46 |
+
trajs_per_obj: all
|
47 |
+
truncate_trajectories_at_success: true
|
48 |
+
max_image_cache_size: all
|
49 |
+
num_dataload_workers: 2
|
50 |
+
num_iterations: 500000
|
51 |
+
eval_episodes: 63
|
52 |
+
log_freq: 1000
|
53 |
+
eval_freq: 5000
|
54 |
+
save_freq: 5000
|
55 |
+
torch_deterministic: true
|
56 |
+
save_backup_ckpts: false
|
57 |
+
logger:
|
58 |
+
workspace: mshab_exps
|
59 |
+
exp_name: rcad-set_table-pick-all
|
60 |
+
clear_out: true
|
61 |
+
tensorboard: true
|
62 |
+
wandb: false
|
63 |
+
model_ckpt: "mshab_checkpoints/dp/set_table/pick/all/policy.pt"
|
dp/set_table/pick/all/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f4cc76b7dddd646d4bab32650cc5b640185d8c15e0d1863c3c930d1a282e927a
|
3 |
+
size 1183023810
|
dp/set_table/place/all/config.yml
ADDED
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: PlaceSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/set_table/place/train/all.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/set_table/place/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- ee_rest
|
25 |
+
- robot_rest
|
26 |
+
- is_static
|
27 |
+
- cumulative_force_within_limit
|
28 |
+
env_kwargs:
|
29 |
+
robot_force_mult: 0.001
|
30 |
+
robot_force_penalty_min: 0.2
|
31 |
+
target_randomization: false
|
32 |
+
algo:
|
33 |
+
name: diffusion_policy
|
34 |
+
lr: 0.0001
|
35 |
+
batch_size: 256
|
36 |
+
obs_horizon: 2
|
37 |
+
act_horizon: 1
|
38 |
+
pred_horizon: 16
|
39 |
+
diffusion_step_embed_dim: 256
|
40 |
+
unet_dims:
|
41 |
+
- 256
|
42 |
+
- 512
|
43 |
+
- 1024
|
44 |
+
n_groups: 8
|
45 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/set_table/place
|
46 |
+
trajs_per_obj: all
|
47 |
+
truncate_trajectories_at_success: true
|
48 |
+
max_image_cache_size: all
|
49 |
+
num_dataload_workers: 2
|
50 |
+
num_iterations: 500000
|
51 |
+
eval_episodes: 63
|
52 |
+
log_freq: 1000
|
53 |
+
eval_freq: 5000
|
54 |
+
save_freq: 5000
|
55 |
+
torch_deterministic: true
|
56 |
+
save_backup_ckpts: false
|
57 |
+
logger:
|
58 |
+
workspace: mshab_exps
|
59 |
+
exp_name: rcad-set_table-place-all
|
60 |
+
clear_out: true
|
61 |
+
tensorboard: true
|
62 |
+
wandb: false
|
63 |
+
model_ckpt: "mshab_checkpoints/dp/set_table/place/all/policy.pt"
|
dp/set_table/place/all/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ec4d3eccffe0ec66d697ecf9a16cc058449bd38db33fe4e96502386a98fc0a5f
|
3 |
+
size 1183023810
|
dp/tidy_house/pick/all/config.yml
ADDED
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: PickSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/tidy_house/pick/train/all.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/tidy_house/pick/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- ee_rest
|
25 |
+
- robot_rest
|
26 |
+
- is_static
|
27 |
+
- cumulative_force_within_limit
|
28 |
+
env_kwargs:
|
29 |
+
robot_force_mult: 0.001
|
30 |
+
robot_force_penalty_min: 0.2
|
31 |
+
target_randomization: false
|
32 |
+
algo:
|
33 |
+
name: diffusion_policy
|
34 |
+
lr: 0.0001
|
35 |
+
batch_size: 256
|
36 |
+
obs_horizon: 2
|
37 |
+
act_horizon: 1
|
38 |
+
pred_horizon: 16
|
39 |
+
diffusion_step_embed_dim: 256
|
40 |
+
unet_dims:
|
41 |
+
- 256
|
42 |
+
- 512
|
43 |
+
- 1024
|
44 |
+
n_groups: 8
|
45 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/tidy_house/pick
|
46 |
+
trajs_per_obj: all
|
47 |
+
truncate_trajectories_at_success: true
|
48 |
+
max_image_cache_size: all
|
49 |
+
num_dataload_workers: 2
|
50 |
+
num_iterations: 500000
|
51 |
+
eval_episodes: 63
|
52 |
+
log_freq: 1000
|
53 |
+
eval_freq: 5000
|
54 |
+
save_freq: 5000
|
55 |
+
torch_deterministic: true
|
56 |
+
save_backup_ckpts: false
|
57 |
+
logger:
|
58 |
+
workspace: mshab_exps
|
59 |
+
exp_name: rcad-tidy_house-pick-all
|
60 |
+
clear_out: true
|
61 |
+
tensorboard: true
|
62 |
+
wandb: false
|
63 |
+
model_ckpt: "mshab_checkpoints/dp/tidy_house/pick/all/policy.pt"
|
dp/tidy_house/pick/all/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bf2907825f88969bcd4f6469ccfa9d5309060847fe37dfe376623420b7f36a02
|
3 |
+
size 1183023810
|
dp/tidy_house/place/all/config.yml
ADDED
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
seed: 0
|
2 |
+
eval_env:
|
3 |
+
env_id: PlaceSubtaskTrain-v0
|
4 |
+
num_envs: 63
|
5 |
+
max_episode_steps: 200
|
6 |
+
make_env: true
|
7 |
+
continuous_task: true
|
8 |
+
cat_state: true
|
9 |
+
cat_pixels: false
|
10 |
+
frame_stack: null
|
11 |
+
stack: 2
|
12 |
+
stationary_base: false
|
13 |
+
stationary_torso: false
|
14 |
+
stationary_head: true
|
15 |
+
task_plan_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/task_plans/tidy_house/place/train/all.json
|
16 |
+
spawn_data_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange/spawn_data/tidy_house/place/train/spawn_data.pt
|
17 |
+
record_video: false
|
18 |
+
debug_video: false
|
19 |
+
debug_video_gen: false
|
20 |
+
save_video_freq: 1
|
21 |
+
info_on_video: false
|
22 |
+
extra_stat_keys:
|
23 |
+
- is_grasped
|
24 |
+
- ee_rest
|
25 |
+
- robot_rest
|
26 |
+
- is_static
|
27 |
+
- cumulative_force_within_limit
|
28 |
+
env_kwargs:
|
29 |
+
robot_force_mult: 0.001
|
30 |
+
robot_force_penalty_min: 0.2
|
31 |
+
target_randomization: false
|
32 |
+
algo:
|
33 |
+
name: diffusion_policy
|
34 |
+
lr: 0.0001
|
35 |
+
batch_size: 256
|
36 |
+
obs_horizon: 2
|
37 |
+
act_horizon: 1
|
38 |
+
pred_horizon: 16
|
39 |
+
diffusion_step_embed_dim: 256
|
40 |
+
unet_dims:
|
41 |
+
- 256
|
42 |
+
- 512
|
43 |
+
- 1024
|
44 |
+
n_groups: 8
|
45 |
+
data_dir_fp: /arshukla-fast-vol-1/ManiSkill/data/scene_datasets/replica_cad_dataset/rearrange-dataset/tidy_house/place
|
46 |
+
trajs_per_obj: all
|
47 |
+
truncate_trajectories_at_success: true
|
48 |
+
max_image_cache_size: all
|
49 |
+
num_dataload_workers: 2
|
50 |
+
num_iterations: 500000
|
51 |
+
eval_episodes: 63
|
52 |
+
log_freq: 1000
|
53 |
+
eval_freq: 5000
|
54 |
+
save_freq: 5000
|
55 |
+
torch_deterministic: true
|
56 |
+
save_backup_ckpts: false
|
57 |
+
logger:
|
58 |
+
workspace: mshab_exps
|
59 |
+
exp_name: rcad-tidy_house-place-all
|
60 |
+
clear_out: true
|
61 |
+
tensorboard: true
|
62 |
+
wandb: false
|
63 |
+
model_ckpt: "mshab_checkpoints/dp/tidy_house/place/all/policy.pt"
|
dp/tidy_house/place/all/policy.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0de89ade0116c5044a0a010974a8434d93f4f0e56f0240353f9388606fa9189c
|
3 |
+
size 1183023810
|