arth-shukla commited on
Commit
0fc83d4
·
verified ·
1 Parent(s): 80dcf18

Upload 20 files

Browse files
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