Jiefeng commited on
Commit
edcfcb8
·
verified ·
1 Parent(s): a7410cb

Upload folder using huggingface_hub

Browse files
.hydra/config.yaml ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ method:
2
+ name: GNFACTOR_BC
3
+ rlbench:
4
+ task_name: hyperTASR_gnfactor
5
+ tasks:
6
+ - close_jar
7
+ - open_drawer
8
+ - sweep_to_dustpan_of_size
9
+ - meat_off_grill
10
+ - turn_tap
11
+ - slide_block_to_color_target
12
+ - put_item_in_drawer
13
+ - reach_and_drag
14
+ - push_buttons
15
+ - stack_blocks
16
+ demo_path: /public/home/group_yangych/yangych/hyperTASR/GNFactor/data/test_data
17
+ episode_length: 25
18
+ cameras:
19
+ - front
20
+ camera_resolution:
21
+ - 128
22
+ - 128
23
+ scene_bounds:
24
+ - -0.3
25
+ - -0.5
26
+ - 0.6
27
+ - 0.7
28
+ - 0.5
29
+ - 1.6
30
+ include_lang_goal_in_obs: true
31
+ time_in_state: true
32
+ headless: true
33
+ framework:
34
+ tensorboard_logging: true
35
+ csv_logging: true
36
+ gpu: 0
37
+ logdir: logs/
38
+ start_seed: 0
39
+ record_every_n: 5
40
+ eval_envs: 5
41
+ eval_from_eps_number: 0
42
+ eval_episodes: 25
43
+ eval_type: missing
44
+ eval_save_metrics: true
45
+ novel_command: None
46
+ cinematic_recorder:
47
+ enabled: false
48
+ camera_resolution:
49
+ - 1280
50
+ - 720
51
+ fps: 30
52
+ rotate_speed: 0.01
53
+ save_path: eval/videos/
.hydra/hydra.yaml ADDED
@@ -0,0 +1,152 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: ${framework.logdir}/${rlbench.task_name}
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ help:
13
+ app_name: ${hydra.job.name}
14
+ header: '${hydra.help.app_name} is powered by Hydra.
15
+
16
+ '
17
+ footer: 'Powered by Hydra (https://hydra.cc)
18
+
19
+ Use --hydra-help to view Hydra specific help
20
+
21
+ '
22
+ template: '${hydra.help.header}
23
+
24
+ == Configuration groups ==
25
+
26
+ Compose your configuration from those groups (group=option)
27
+
28
+
29
+ $APP_CONFIG_GROUPS
30
+
31
+
32
+ == Config ==
33
+
34
+ Override anything in the config (foo.bar=value)
35
+
36
+
37
+ $CONFIG
38
+
39
+
40
+ ${hydra.help.footer}
41
+
42
+ '
43
+ hydra_help:
44
+ template: 'Hydra (${hydra.runtime.version})
45
+
46
+ See https://hydra.cc for more info.
47
+
48
+
49
+ == Flags ==
50
+
51
+ $FLAGS_HELP
52
+
53
+
54
+ == Configuration groups ==
55
+
56
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
57
+ to command line)
58
+
59
+
60
+ $HYDRA_CONFIG_GROUPS
61
+
62
+
63
+ Use ''--cfg hydra'' to Show the Hydra config.
64
+
65
+ '
66
+ hydra_help: ???
67
+ hydra_logging:
68
+ version: 1
69
+ formatters:
70
+ simple:
71
+ format: '[%(asctime)s][HYDRA] %(message)s'
72
+ handlers:
73
+ console:
74
+ class: logging.StreamHandler
75
+ formatter: simple
76
+ stream: ext://sys.stdout
77
+ root:
78
+ level: INFO
79
+ handlers:
80
+ - console
81
+ loggers:
82
+ logging_example:
83
+ level: DEBUG
84
+ disable_existing_loggers: false
85
+ job_logging:
86
+ version: 1
87
+ formatters:
88
+ simple:
89
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
90
+ handlers:
91
+ console:
92
+ class: logging.StreamHandler
93
+ formatter: simple
94
+ stream: ext://sys.stdout
95
+ file:
96
+ class: logging.FileHandler
97
+ formatter: simple
98
+ filename: ${hydra.job.name}.log
99
+ root:
100
+ level: INFO
101
+ handlers:
102
+ - console
103
+ - file
104
+ disable_existing_loggers: false
105
+ env: {}
106
+ searchpath: []
107
+ callbacks: {}
108
+ output_subdir: .hydra
109
+ overrides:
110
+ hydra: []
111
+ task:
112
+ - method.name=GNFACTOR_BC
113
+ - rlbench.task_name=hyperTASR_gnfactor
114
+ - rlbench.demo_path=/public/home/group_yangych/yangych/hyperTASR/GNFactor/data/test_data
115
+ - framework.start_seed=0
116
+ job:
117
+ name: eval
118
+ override_dirname: framework.start_seed=0,method.name=GNFACTOR_BC,rlbench.demo_path=/public/home/group_yangych/yangych/hyperTASR/GNFactor/data/test_data,rlbench.task_name=hyperTASR_gnfactor
119
+ id: ???
120
+ num: ???
121
+ config_name: eval
122
+ env_set: {}
123
+ env_copy: []
124
+ config:
125
+ override_dirname:
126
+ kv_sep: '='
127
+ item_sep: ','
128
+ exclude_keys: []
129
+ runtime:
130
+ version: 1.1.0
131
+ cwd: /public/home/group_yangych/yangych/hyperTASR/GNFactor/GNFactor
132
+ config_sources:
133
+ - path: hydra.conf
134
+ schema: pkg
135
+ provider: hydra
136
+ - path: /public/home/group_yangych/yangych/hyperTASR/GNFactor/GNFactor/conf
137
+ schema: file
138
+ provider: main
139
+ - path: ''
140
+ schema: structured
141
+ provider: schema
142
+ choices:
143
+ hydra/env: default
144
+ hydra/callbacks: null
145
+ hydra/job_logging: default
146
+ hydra/hydra_logging: default
147
+ hydra/hydra_help: default
148
+ hydra/help: default
149
+ hydra/sweeper: basic
150
+ hydra/launcher: basic
151
+ hydra/output: default
152
+ verbose: false
.hydra/overrides.yaml ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ - method.name=GNFACTOR_BC
2
+ - rlbench.task_name=hyperTASR_gnfactor
3
+ - rlbench.demo_path=/public/home/group_yangych/yangych/hyperTASR/GNFactor/data/test_data
4
+ - framework.start_seed=0
eval.log ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [2025-10-13 01:14:35,367][root][INFO] - Using env device cuda:0.
2
+ [2025-10-13 01:14:35,643][root][INFO] - Evaluating seed 0.
3
+ [2025-10-13 17:43:39,297][root][INFO] - Using env device cuda:0.
4
+ [2025-10-13 17:43:39,522][root][INFO] - Evaluating seed 0.
5
+ [2025-10-13 17:44:40,164][root][INFO] - Using env device cuda:0.
6
+ [2025-10-13 17:44:40,261][root][INFO] - Evaluating seed 0.
7
+ [2025-10-14 00:16:42,525][root][INFO] - Using env device cuda:0.
8
+ [2025-10-14 00:16:42,627][root][INFO] - Evaluating seed 0.
9
+ [2025-10-14 00:59:17,859][root][INFO] - Using env device cuda:0.
10
+ [2025-10-14 00:59:18,014][root][INFO] - Evaluating seed 0.
11
+ [2025-10-16 22:42:26,198][root][INFO] - Using env device cuda:0.
12
+ [2025-10-16 22:42:26,446][root][INFO] - Evaluating seed 0.
13
+ [2025-10-16 22:48:56,436][root][INFO] - Using env device cuda:0.
14
+ [2025-10-16 22:48:56,475][root][INFO] - Evaluating seed 0.
15
+ [2025-10-16 22:50:37,508][root][INFO] - Using env device cuda:0.
16
+ [2025-10-16 22:50:37,547][root][INFO] - Evaluating seed 0.
17
+ [2025-10-16 22:55:27,278][root][INFO] - Using env device cuda:0.
18
+ [2025-10-16 22:55:27,320][root][INFO] - Evaluating seed 0.
seed0/config.yaml ADDED
@@ -0,0 +1,196 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ method:
2
+ name: GNFACTOR_BC
3
+ use_neural_rendering: false
4
+ num_view_for_nerf: 20
5
+ language_model: CLIP
6
+ language_model_dim: 512
7
+ image_crop_size: 64
8
+ bounds_offset:
9
+ - 0.15
10
+ voxel_sizes:
11
+ - 100
12
+ include_prev_layer: false
13
+ num_latents: 2048
14
+ latent_dim: 512
15
+ transformer_depth: 6
16
+ transformer_iterations: 1
17
+ cross_heads: 1
18
+ cross_dim_head: 64
19
+ latent_heads: 8
20
+ latent_dim_head: 64
21
+ pos_encoding_with_lang: true
22
+ conv_downsample: true
23
+ lang_fusion_type: seq
24
+ voxel_patch_size: 5
25
+ voxel_patch_stride: 5
26
+ final_dim: 128
27
+ input_dropout: 0.1
28
+ attn_dropout: 0.1
29
+ decoder_dropout: 0.0
30
+ lr: 0.0005
31
+ lr_scheduler: false
32
+ num_warmup_steps: 3000
33
+ optimizer: lamb
34
+ lambda_weight_l2: 1.0e-06
35
+ trans_loss_weight: 1.0
36
+ rot_loss_weight: 1.0
37
+ grip_loss_weight: 1.0
38
+ collision_loss_weight: 1.0
39
+ rotation_resolution: 5
40
+ activation: lrelu
41
+ norm: None
42
+ crop_augmentation: true
43
+ transform_augmentation:
44
+ apply_se3: true
45
+ aug_xyz:
46
+ - 0.125
47
+ - 0.125
48
+ - 0.125
49
+ aug_rpy:
50
+ - 0.0
51
+ - 0.0
52
+ - 45.0
53
+ aug_rot_resolution: ${method.rotation_resolution}
54
+ demo_augmentation: true
55
+ demo_augmentation_every_n: 10
56
+ no_skip_connection: false
57
+ no_perceiver: false
58
+ no_language: false
59
+ keypoint_method: heuristic
60
+ use_wandb: true
61
+ neural_renderer:
62
+ foundation_model_name: diffusion
63
+ d_embed: 512
64
+ d_latent: 128
65
+ use_multi_scale_voxel: false
66
+ d_multi_scale_latent: 266
67
+ use_depth_supervision: false
68
+ lambda_nerf: 0.01
69
+ lambda_embed: 0.01
70
+ lambda_rgb: 1.0
71
+ ray_chunk_size: 512
72
+ d_lang: 128
73
+ voxel_shape: 100
74
+ share_mlp: true
75
+ image_width: 128
76
+ image_height: 128
77
+ z_near: 0.1
78
+ z_far: 4.0
79
+ coordinate_bounds:
80
+ - -0.3
81
+ - -0.5
82
+ - 0.6
83
+ - 0.7
84
+ - 0.5
85
+ - 1.6
86
+ regress_coord: false
87
+ regress_attention: false
88
+ ret_last_feat: false
89
+ use_code: true
90
+ use_code_viewdirs: false
91
+ use_xyz: true
92
+ n_coarse: 64
93
+ n_fine: 32
94
+ n_fine_depth: 16
95
+ white_bkgd: false
96
+ lindisp: false
97
+ normalize_z: false
98
+ canon_xyz: true
99
+ use_viewdirs: true
100
+ eval_batch_size: 4096
101
+ noise_std: 0.0
102
+ depth_std: 0.001
103
+ mlp:
104
+ n_blocks: 5
105
+ d_hidden: 512
106
+ combine_layer: 3
107
+ combine_type: average
108
+ beta: 0.0
109
+ use_spade: false
110
+ code:
111
+ num_freqs: 6
112
+ freq_factor: 1.5
113
+ include_input: true
114
+ ddp:
115
+ master_addr: localhost
116
+ master_port: 12345
117
+ num_devices: 2
118
+ device_id: 0
119
+ rlbench:
120
+ task_name: GNFACTOR_BC_20250109
121
+ tasks:
122
+ - close_jar
123
+ - open_drawer
124
+ - sweep_to_dustpan_of_size
125
+ - meat_off_grill
126
+ - turn_tap
127
+ - slide_block_to_color_target
128
+ - put_item_in_drawer
129
+ - reach_and_drag
130
+ - push_buttons
131
+ - stack_blocks
132
+ demos: 20
133
+ num_view_for_nerf: 21
134
+ demo_path: /public/home/group_yangych/yangych/GNF/GNFactor_hyperopt/data/train_data
135
+ episode_length: 15
136
+ cameras:
137
+ - front
138
+ camera_resolution:
139
+ - 128
140
+ - 128
141
+ scene_bounds:
142
+ - -0.3
143
+ - -0.5
144
+ - 0.6
145
+ - 0.7
146
+ - 0.5
147
+ - 1.6
148
+ include_lang_goal_in_obs: true
149
+ headless: true
150
+ replay:
151
+ batch_size: 1
152
+ timesteps: 1
153
+ prioritisation: false
154
+ task_uniform: true
155
+ use_disk: true
156
+ path: /public/home/group_yangych/yangych/GNF/GNFactor_hyperopt/replay/GNFACTOR_BC_20250109
157
+ max_parallel_processes: 8
158
+ evaluation:
159
+ eval_freq: 10000
160
+ eval_episodes: 25
161
+ time_in_state: true
162
+ record_every_n: 10000
163
+ episode_length: 15
164
+ gpu: 0
165
+ framework:
166
+ useHyper:
167
+ - false
168
+ - false
169
+ - false
170
+ - true
171
+ - true
172
+ - true
173
+ - true
174
+ use_online_evaluation: false
175
+ log_freq: 100
176
+ save_freq: 10000
177
+ train_envs: 1
178
+ replay_ratio: ${replay.batch_size}
179
+ transitions_before_train: 200
180
+ tensorboard_logging: false
181
+ csv_logging: true
182
+ training_iterations: 100010
183
+ gpu: 0
184
+ env_gpu: 0
185
+ logdir: logs/
186
+ logging_level: 20
187
+ seeds: 1
188
+ start_seed: 0
189
+ load_existing_weights: false
190
+ num_weights_to_keep: 60
191
+ num_workers: 0
192
+ record_every_n: 5
193
+ use_wandb: true
194
+ wandb_project: gnfactor
195
+ wandb_group: GNFACTOR_BC_20250109
196
+ seed: 0
seed0/eval_data.csv ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ step,eval_envs/return/close_jar,eval_envs/length/close_jar,eval_envs/total_transitions/close_jar,eval_envs/return/open_drawer,eval_envs/length/open_drawer,eval_envs/total_transitions/open_drawer,eval_envs/return/sweep_to_dustpan_of_size,eval_envs/length/sweep_to_dustpan_of_size,eval_envs/total_transitions/sweep_to_dustpan_of_size,eval_envs/return/meat_off_grill,eval_envs/length/meat_off_grill,eval_envs/total_transitions/meat_off_grill,eval_envs/return/turn_tap,eval_envs/length/turn_tap,eval_envs/total_transitions/turn_tap,eval_envs/return/slide_block_to_color_target,eval_envs/length/slide_block_to_color_target,eval_envs/total_transitions/slide_block_to_color_target,eval_envs/return/put_item_in_drawer,eval_envs/length/put_item_in_drawer,eval_envs/total_transitions/put_item_in_drawer,eval_envs/return/reach_and_drag,eval_envs/length/reach_and_drag,eval_envs/total_transitions/reach_and_drag,eval_envs/return/push_buttons,eval_envs/length/push_buttons,eval_envs/total_transitions/push_buttons,eval_envs/return/stack_blocks,eval_envs/length/stack_blocks,eval_envs/total_transitions/stack_blocks
2
+ 200000,24.0,19.92,498,64.0,10.36,757,40.0,17.76,1201,76.0,11.52,1489,60.0,10.24,1745,28.0,16.88,2167,24.0,20.68,2684,88.0,9.2,2914,32.0,18.28,3371,8.0,21.8,3916
seed0/weights/200000/QAttentionAgent_layer0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e62437028b0e8846878b7cb2a3c3afdce35e93db2b3bd34c2bab4fb9a9a7801
3
+ size 396972411