Zihao Li commited on
Commit
863af97
·
verified ·
1 Parent(s): 98512c9

Upload checkpoint step 60000

Browse files

Run dir: /workspace/corl/outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716
Checkpoint dir: /workspace/corl/outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000
Mode: full
Training step: 60000

Files changed (11) hide show
  1. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/config.json +178 -0
  2. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/model.safetensors +3 -0
  3. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/policy_postprocessor.json +32 -0
  4. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  5. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/policy_preprocessor.json +149 -0
  6. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  7. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/train_config.json +460 -0
  8. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/training_state/optimizer_param_groups.json +227 -0
  9. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/training_state/optimizer_state.safetensors +3 -0
  10. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/training_state/rng_state.safetensors +3 -0
  11. outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/training_state/training_step.json +3 -0
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/config.json ADDED
@@ -0,0 +1,178 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "streaming_act",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.state": {
6
+ "type": "STATE",
7
+ "shape": [
8
+ 17
9
+ ]
10
+ },
11
+ "observation.images.realsense_top": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 224,
16
+ 224
17
+ ]
18
+ },
19
+ "observation.images.realsense_left": {
20
+ "type": "VISUAL",
21
+ "shape": [
22
+ 3,
23
+ 224,
24
+ 224
25
+ ]
26
+ },
27
+ "observation.images.realsense_right": {
28
+ "type": "VISUAL",
29
+ "shape": [
30
+ 3,
31
+ 224,
32
+ 224
33
+ ]
34
+ },
35
+ "observation.path_signature": {
36
+ "type": "STATE",
37
+ "shape": [
38
+ 5219
39
+ ]
40
+ },
41
+ "observation.delta_signature": {
42
+ "type": "STATE",
43
+ "shape": [
44
+ 5219
45
+ ]
46
+ },
47
+ "observation.prefix_state": {
48
+ "type": "STATE",
49
+ "shape": [
50
+ 24,
51
+ 17
52
+ ]
53
+ },
54
+ "observation.prefix_mask": {
55
+ "type": "STATE",
56
+ "shape": [
57
+ 24
58
+ ]
59
+ },
60
+ "observation.prefix_path_signature": {
61
+ "type": "STATE",
62
+ "shape": [
63
+ 24,
64
+ 5219
65
+ ]
66
+ },
67
+ "observation.prefix_delta_signature": {
68
+ "type": "STATE",
69
+ "shape": [
70
+ 24,
71
+ 5219
72
+ ]
73
+ },
74
+ "observation.prefix_images.realsense_top": {
75
+ "type": "VISUAL",
76
+ "shape": [
77
+ 24,
78
+ 3,
79
+ 224,
80
+ 224
81
+ ]
82
+ },
83
+ "observation.prefix_images.realsense_left": {
84
+ "type": "VISUAL",
85
+ "shape": [
86
+ 24,
87
+ 3,
88
+ 224,
89
+ 224
90
+ ]
91
+ },
92
+ "observation.prefix_images.realsense_right": {
93
+ "type": "VISUAL",
94
+ "shape": [
95
+ 24,
96
+ 3,
97
+ 224,
98
+ 224
99
+ ]
100
+ }
101
+ },
102
+ "output_features": {
103
+ "action": {
104
+ "type": "ACTION",
105
+ "shape": [
106
+ 17
107
+ ]
108
+ }
109
+ },
110
+ "device": "cuda",
111
+ "use_amp": true,
112
+ "use_peft": false,
113
+ "push_to_hub": false,
114
+ "repo_id": null,
115
+ "private": null,
116
+ "tags": null,
117
+ "license": null,
118
+ "pretrained_path": null,
119
+ "chunk_size": 100,
120
+ "n_action_steps": 100,
121
+ "normalization_mapping": {
122
+ "VISUAL": "MEAN_STD",
123
+ "STATE": "MEAN_STD",
124
+ "ACTION": "MEAN_STD"
125
+ },
126
+ "vision_backbone": "resnet18",
127
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
128
+ "replace_final_stride_with_dilation": false,
129
+ "pre_norm": false,
130
+ "dim_model": 512,
131
+ "n_heads": 8,
132
+ "dim_feedforward": 3200,
133
+ "feedforward_activation": "relu",
134
+ "n_encoder_layers": 4,
135
+ "n_decoder_layers": 1,
136
+ "use_vae": true,
137
+ "latent_dim": 32,
138
+ "n_vae_encoder_layers": 4,
139
+ "use_first_frame_anchor": false,
140
+ "use_path_signature": true,
141
+ "use_prefix_sequence_training": true,
142
+ "prefix_train_max_steps": 24,
143
+ "prefix_frame_stride": 6,
144
+ "prefix_pad_value": 0.0,
145
+ "use_visual_prefix_memory": true,
146
+ "use_delta_signature": true,
147
+ "use_signature_conditioned_visual_prefix_memory": false,
148
+ "use_signature_indexed_slot_memory": true,
149
+ "use_memory_conditioned_encoder_film": true,
150
+ "num_memory_slots": 1,
151
+ "slot_memory_num_slots": 4,
152
+ "slot_memory_routing_hidden_dim": 512,
153
+ "slot_memory_identity_scale": 0.5,
154
+ "slot_memory_routing_temperature": 1.0,
155
+ "slot_memory_use_delta_routing": true,
156
+ "slot_memory_use_softmax_routing": true,
157
+ "slot_memory_use_readout_pooling": true,
158
+ "slot_memory_balance_loss_coef": 0.02,
159
+ "slot_memory_entropy_loss_coef": 0.003,
160
+ "slot_memory_consistency_loss_coef": 0.0,
161
+ "history_length": 2556,
162
+ "signature_dim": 5219,
163
+ "signature_depth": 3,
164
+ "signature_hidden_dim": 512,
165
+ "signature_dropout": 0.1,
166
+ "pre_normalized_observation_keys": [
167
+ "observation.path_signature",
168
+ "observation.delta_signature",
169
+ "observation.prefix_path_signature",
170
+ "observation.prefix_delta_signature"
171
+ ],
172
+ "temporal_ensemble_coeff": null,
173
+ "dropout": 0.1,
174
+ "kl_weight": 10.0,
175
+ "optimizer_lr": 1e-05,
176
+ "optimizer_weight_decay": 0.0001,
177
+ "optimizer_lr_backbone": 1e-05
178
+ }
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87bcec6c3586598101d33f0c0019b59a882d04ee28f34ac9d818c60d7cff1521
3
+ size 257565824
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/policy_postprocessor.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "eps": 1e-08,
8
+ "features": {
9
+ "action": {
10
+ "type": "ACTION",
11
+ "shape": [
12
+ 17
13
+ ]
14
+ }
15
+ },
16
+ "norm_map": {
17
+ "VISUAL": "MEAN_STD",
18
+ "STATE": "MEAN_STD",
19
+ "ACTION": "MEAN_STD"
20
+ }
21
+ },
22
+ "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
23
+ },
24
+ {
25
+ "registry_name": "device_processor",
26
+ "config": {
27
+ "device": "cpu",
28
+ "float_dtype": null
29
+ }
30
+ }
31
+ ]
32
+ }
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fbf26ca6ebff3c8db3db8788f02aae78716ae3f23a6a9166fea341912519a85
3
+ size 351080
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/policy_preprocessor.json ADDED
@@ -0,0 +1,149 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "rename_observations_processor",
6
+ "config": {
7
+ "rename_map": {}
8
+ }
9
+ },
10
+ {
11
+ "registry_name": "to_batch_processor",
12
+ "config": {}
13
+ },
14
+ {
15
+ "registry_name": "device_processor",
16
+ "config": {
17
+ "device": "cuda",
18
+ "float_dtype": null
19
+ }
20
+ },
21
+ {
22
+ "registry_name": "normalizer_processor",
23
+ "config": {
24
+ "eps": 1e-08,
25
+ "features": {
26
+ "observation.state": {
27
+ "type": "STATE",
28
+ "shape": [
29
+ 17
30
+ ]
31
+ },
32
+ "observation.images.realsense_top": {
33
+ "type": "VISUAL",
34
+ "shape": [
35
+ 3,
36
+ 224,
37
+ 224
38
+ ]
39
+ },
40
+ "observation.images.realsense_left": {
41
+ "type": "VISUAL",
42
+ "shape": [
43
+ 3,
44
+ 224,
45
+ 224
46
+ ]
47
+ },
48
+ "observation.images.realsense_right": {
49
+ "type": "VISUAL",
50
+ "shape": [
51
+ 3,
52
+ 224,
53
+ 224
54
+ ]
55
+ },
56
+ "observation.path_signature": {
57
+ "type": "STATE",
58
+ "shape": [
59
+ 5219
60
+ ]
61
+ },
62
+ "observation.delta_signature": {
63
+ "type": "STATE",
64
+ "shape": [
65
+ 5219
66
+ ]
67
+ },
68
+ "observation.prefix_state": {
69
+ "type": "STATE",
70
+ "shape": [
71
+ 24,
72
+ 17
73
+ ]
74
+ },
75
+ "observation.prefix_mask": {
76
+ "type": "STATE",
77
+ "shape": [
78
+ 24
79
+ ]
80
+ },
81
+ "observation.prefix_path_signature": {
82
+ "type": "STATE",
83
+ "shape": [
84
+ 24,
85
+ 5219
86
+ ]
87
+ },
88
+ "observation.prefix_delta_signature": {
89
+ "type": "STATE",
90
+ "shape": [
91
+ 24,
92
+ 5219
93
+ ]
94
+ },
95
+ "observation.prefix_images.realsense_top": {
96
+ "type": "VISUAL",
97
+ "shape": [
98
+ 24,
99
+ 3,
100
+ 224,
101
+ 224
102
+ ]
103
+ },
104
+ "observation.prefix_images.realsense_left": {
105
+ "type": "VISUAL",
106
+ "shape": [
107
+ 24,
108
+ 3,
109
+ 224,
110
+ 224
111
+ ]
112
+ },
113
+ "observation.prefix_images.realsense_right": {
114
+ "type": "VISUAL",
115
+ "shape": [
116
+ 24,
117
+ 3,
118
+ 224,
119
+ 224
120
+ ]
121
+ },
122
+ "action": {
123
+ "type": "ACTION",
124
+ "shape": [
125
+ 17
126
+ ]
127
+ }
128
+ },
129
+ "norm_map": {
130
+ "VISUAL": "MEAN_STD",
131
+ "STATE": "MEAN_STD",
132
+ "ACTION": "MEAN_STD"
133
+ },
134
+ "normalize_observation_keys": [
135
+ "observation.images.realsense_left",
136
+ "observation.images.realsense_right",
137
+ "observation.images.realsense_top",
138
+ "observation.prefix_images.realsense_left",
139
+ "observation.prefix_images.realsense_right",
140
+ "observation.prefix_images.realsense_top",
141
+ "observation.prefix_mask",
142
+ "observation.prefix_state",
143
+ "observation.state"
144
+ ]
145
+ },
146
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
147
+ }
148
+ ]
149
+ }
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fbf26ca6ebff3c8db3db8788f02aae78716ae3f23a6a9166fea341912519a85
3
+ size 351080
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/pretrained_model/train_config.json ADDED
@@ -0,0 +1,460 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "zeno-ai/BookOriginRelocation",
4
+ "root": "/workspace/corl/data/zeno-ai/BookOriginRelocation",
5
+ "episodes": [
6
+ 0,
7
+ 1,
8
+ 2,
9
+ 3,
10
+ 4,
11
+ 5,
12
+ 6,
13
+ 7,
14
+ 8,
15
+ 9,
16
+ 10,
17
+ 11,
18
+ 12,
19
+ 13,
20
+ 14,
21
+ 15,
22
+ 16,
23
+ 17,
24
+ 18,
25
+ 19,
26
+ 20,
27
+ 21,
28
+ 22,
29
+ 23,
30
+ 24,
31
+ 25,
32
+ 26,
33
+ 27,
34
+ 28,
35
+ 29,
36
+ 30,
37
+ 31,
38
+ 32,
39
+ 33,
40
+ 34,
41
+ 35,
42
+ 36,
43
+ 37,
44
+ 38,
45
+ 39,
46
+ 40,
47
+ 41,
48
+ 42,
49
+ 43,
50
+ 44,
51
+ 45,
52
+ 46,
53
+ 47,
54
+ 48,
55
+ 49,
56
+ 50,
57
+ 51,
58
+ 52,
59
+ 53,
60
+ 54,
61
+ 55,
62
+ 56,
63
+ 57,
64
+ 58,
65
+ 59,
66
+ 60,
67
+ 61,
68
+ 62,
69
+ 63,
70
+ 64,
71
+ 65,
72
+ 66,
73
+ 67,
74
+ 68,
75
+ 69,
76
+ 70,
77
+ 71,
78
+ 72,
79
+ 73,
80
+ 74,
81
+ 75,
82
+ 76,
83
+ 77,
84
+ 78,
85
+ 79,
86
+ 80,
87
+ 81,
88
+ 82,
89
+ 83,
90
+ 84,
91
+ 85,
92
+ 86,
93
+ 87,
94
+ 88,
95
+ 89,
96
+ 90,
97
+ 91,
98
+ 92,
99
+ 93,
100
+ 94,
101
+ 95,
102
+ 96,
103
+ 97,
104
+ 98,
105
+ 99,
106
+ 100,
107
+ 101,
108
+ 102,
109
+ 103,
110
+ 104,
111
+ 105,
112
+ 106,
113
+ 107,
114
+ 108,
115
+ 109,
116
+ 110,
117
+ 111,
118
+ 112,
119
+ 113,
120
+ 114,
121
+ 115,
122
+ 116,
123
+ 117,
124
+ 118,
125
+ 119,
126
+ 120,
127
+ 121,
128
+ 122,
129
+ 123,
130
+ 124,
131
+ 125,
132
+ 126,
133
+ 127,
134
+ 128,
135
+ 129,
136
+ 130,
137
+ 131,
138
+ 132,
139
+ 133,
140
+ 134,
141
+ 135,
142
+ 136,
143
+ 137,
144
+ 138,
145
+ 139,
146
+ 140,
147
+ 141,
148
+ 142,
149
+ 143,
150
+ 144,
151
+ 145,
152
+ 146,
153
+ 147,
154
+ 148,
155
+ 149
156
+ ],
157
+ "image_transforms": {
158
+ "enable": false,
159
+ "max_num_transforms": 3,
160
+ "random_order": false,
161
+ "tfs": {
162
+ "brightness": {
163
+ "weight": 1.0,
164
+ "type": "ColorJitter",
165
+ "kwargs": {
166
+ "brightness": [
167
+ 0.8,
168
+ 1.2
169
+ ]
170
+ }
171
+ },
172
+ "contrast": {
173
+ "weight": 1.0,
174
+ "type": "ColorJitter",
175
+ "kwargs": {
176
+ "contrast": [
177
+ 0.8,
178
+ 1.2
179
+ ]
180
+ }
181
+ },
182
+ "saturation": {
183
+ "weight": 1.0,
184
+ "type": "ColorJitter",
185
+ "kwargs": {
186
+ "saturation": [
187
+ 0.5,
188
+ 1.5
189
+ ]
190
+ }
191
+ },
192
+ "hue": {
193
+ "weight": 1.0,
194
+ "type": "ColorJitter",
195
+ "kwargs": {
196
+ "hue": [
197
+ -0.05,
198
+ 0.05
199
+ ]
200
+ }
201
+ },
202
+ "sharpness": {
203
+ "weight": 1.0,
204
+ "type": "SharpnessJitter",
205
+ "kwargs": {
206
+ "sharpness": [
207
+ 0.5,
208
+ 1.5
209
+ ]
210
+ }
211
+ },
212
+ "affine": {
213
+ "weight": 1.0,
214
+ "type": "RandomAffine",
215
+ "kwargs": {
216
+ "degrees": [
217
+ -5.0,
218
+ 5.0
219
+ ],
220
+ "translate": [
221
+ 0.05,
222
+ 0.05
223
+ ]
224
+ }
225
+ }
226
+ }
227
+ },
228
+ "revision": null,
229
+ "use_imagenet_stats": false,
230
+ "video_backend": "torchcodec",
231
+ "streaming": false
232
+ },
233
+ "env": null,
234
+ "policy": {
235
+ "type": "streaming_act",
236
+ "n_obs_steps": 1,
237
+ "input_features": {
238
+ "observation.state": {
239
+ "type": "STATE",
240
+ "shape": [
241
+ 17
242
+ ]
243
+ },
244
+ "observation.images.realsense_top": {
245
+ "type": "VISUAL",
246
+ "shape": [
247
+ 3,
248
+ 224,
249
+ 224
250
+ ]
251
+ },
252
+ "observation.images.realsense_left": {
253
+ "type": "VISUAL",
254
+ "shape": [
255
+ 3,
256
+ 224,
257
+ 224
258
+ ]
259
+ },
260
+ "observation.images.realsense_right": {
261
+ "type": "VISUAL",
262
+ "shape": [
263
+ 3,
264
+ 224,
265
+ 224
266
+ ]
267
+ },
268
+ "observation.path_signature": {
269
+ "type": "STATE",
270
+ "shape": [
271
+ 5219
272
+ ]
273
+ },
274
+ "observation.delta_signature": {
275
+ "type": "STATE",
276
+ "shape": [
277
+ 5219
278
+ ]
279
+ },
280
+ "observation.prefix_state": {
281
+ "type": "STATE",
282
+ "shape": [
283
+ 24,
284
+ 17
285
+ ]
286
+ },
287
+ "observation.prefix_mask": {
288
+ "type": "STATE",
289
+ "shape": [
290
+ 24
291
+ ]
292
+ },
293
+ "observation.prefix_path_signature": {
294
+ "type": "STATE",
295
+ "shape": [
296
+ 24,
297
+ 5219
298
+ ]
299
+ },
300
+ "observation.prefix_delta_signature": {
301
+ "type": "STATE",
302
+ "shape": [
303
+ 24,
304
+ 5219
305
+ ]
306
+ },
307
+ "observation.prefix_images.realsense_top": {
308
+ "type": "VISUAL",
309
+ "shape": [
310
+ 24,
311
+ 3,
312
+ 224,
313
+ 224
314
+ ]
315
+ },
316
+ "observation.prefix_images.realsense_left": {
317
+ "type": "VISUAL",
318
+ "shape": [
319
+ 24,
320
+ 3,
321
+ 224,
322
+ 224
323
+ ]
324
+ },
325
+ "observation.prefix_images.realsense_right": {
326
+ "type": "VISUAL",
327
+ "shape": [
328
+ 24,
329
+ 3,
330
+ 224,
331
+ 224
332
+ ]
333
+ }
334
+ },
335
+ "output_features": {
336
+ "action": {
337
+ "type": "ACTION",
338
+ "shape": [
339
+ 17
340
+ ]
341
+ }
342
+ },
343
+ "device": "cuda",
344
+ "use_amp": true,
345
+ "use_peft": false,
346
+ "push_to_hub": false,
347
+ "repo_id": null,
348
+ "private": null,
349
+ "tags": null,
350
+ "license": null,
351
+ "pretrained_path": null,
352
+ "chunk_size": 100,
353
+ "n_action_steps": 100,
354
+ "normalization_mapping": {
355
+ "VISUAL": "MEAN_STD",
356
+ "STATE": "MEAN_STD",
357
+ "ACTION": "MEAN_STD"
358
+ },
359
+ "vision_backbone": "resnet18",
360
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
361
+ "replace_final_stride_with_dilation": false,
362
+ "pre_norm": false,
363
+ "dim_model": 512,
364
+ "n_heads": 8,
365
+ "dim_feedforward": 3200,
366
+ "feedforward_activation": "relu",
367
+ "n_encoder_layers": 4,
368
+ "n_decoder_layers": 1,
369
+ "use_vae": true,
370
+ "latent_dim": 32,
371
+ "n_vae_encoder_layers": 4,
372
+ "use_first_frame_anchor": false,
373
+ "use_path_signature": true,
374
+ "use_prefix_sequence_training": true,
375
+ "prefix_train_max_steps": 24,
376
+ "prefix_frame_stride": 6,
377
+ "prefix_pad_value": 0.0,
378
+ "use_visual_prefix_memory": true,
379
+ "use_delta_signature": true,
380
+ "use_signature_conditioned_visual_prefix_memory": false,
381
+ "use_signature_indexed_slot_memory": true,
382
+ "use_memory_conditioned_encoder_film": true,
383
+ "num_memory_slots": 1,
384
+ "slot_memory_num_slots": 4,
385
+ "slot_memory_routing_hidden_dim": 512,
386
+ "slot_memory_identity_scale": 0.5,
387
+ "slot_memory_routing_temperature": 1.0,
388
+ "slot_memory_use_delta_routing": true,
389
+ "slot_memory_use_softmax_routing": true,
390
+ "slot_memory_use_readout_pooling": true,
391
+ "slot_memory_balance_loss_coef": 0.02,
392
+ "slot_memory_entropy_loss_coef": 0.003,
393
+ "slot_memory_consistency_loss_coef": 0.0,
394
+ "history_length": 2556,
395
+ "signature_dim": 5219,
396
+ "signature_depth": 3,
397
+ "signature_hidden_dim": 512,
398
+ "signature_dropout": 0.1,
399
+ "pre_normalized_observation_keys": [
400
+ "observation.path_signature",
401
+ "observation.delta_signature",
402
+ "observation.prefix_path_signature",
403
+ "observation.prefix_delta_signature"
404
+ ],
405
+ "temporal_ensemble_coeff": null,
406
+ "dropout": 0.1,
407
+ "kl_weight": 10.0,
408
+ "optimizer_lr": 1e-05,
409
+ "optimizer_weight_decay": 0.0001,
410
+ "optimizer_lr_backbone": 1e-05
411
+ },
412
+ "output_dir": "/workspace/corl/outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716",
413
+ "job_name": "streaming-act-prism-4x4090-s42-20260504_103716",
414
+ "resume": false,
415
+ "seed": 42,
416
+ "cudnn_deterministic": false,
417
+ "num_workers": 8,
418
+ "batch_size": 8,
419
+ "steps": 360000,
420
+ "eval_freq": -1,
421
+ "log_freq": 100,
422
+ "tolerance_s": 0.0001,
423
+ "save_checkpoint": true,
424
+ "save_freq": 30000,
425
+ "use_policy_training_preset": true,
426
+ "optimizer": {
427
+ "type": "adamw",
428
+ "lr": 1e-05,
429
+ "weight_decay": 0.0001,
430
+ "grad_clip_norm": 10.0,
431
+ "betas": [
432
+ 0.9,
433
+ 0.999
434
+ ],
435
+ "eps": 1e-08
436
+ },
437
+ "scheduler": null,
438
+ "eval": {
439
+ "n_episodes": 50,
440
+ "batch_size": 50,
441
+ "use_async_envs": false
442
+ },
443
+ "wandb": {
444
+ "enable": true,
445
+ "disable_artifact": false,
446
+ "project": "BookOriginRelocation",
447
+ "entity": null,
448
+ "notes": null,
449
+ "run_id": "c3v2oxss",
450
+ "mode": "online"
451
+ },
452
+ "peft": null,
453
+ "use_rabc": false,
454
+ "rabc_progress_path": null,
455
+ "rabc_kappa": 0.01,
456
+ "rabc_epsilon": 1e-06,
457
+ "rabc_head_mode": "sparse",
458
+ "rename_map": {},
459
+ "checkpoint_path": null
460
+ }
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/training_state/optimizer_param_groups.json ADDED
@@ -0,0 +1,227 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "lr": 1e-05,
4
+ "betas": [
5
+ 0.9,
6
+ 0.999
7
+ ],
8
+ "eps": 1e-08,
9
+ "weight_decay": 0.0001,
10
+ "amsgrad": false,
11
+ "maximize": false,
12
+ "foreach": null,
13
+ "capturable": false,
14
+ "differentiable": false,
15
+ "fused": null,
16
+ "decoupled_weight_decay": true,
17
+ "params": [
18
+ 0,
19
+ 1,
20
+ 2,
21
+ 3,
22
+ 4,
23
+ 5,
24
+ 6,
25
+ 7,
26
+ 8,
27
+ 9,
28
+ 10,
29
+ 11,
30
+ 12,
31
+ 13,
32
+ 14,
33
+ 15,
34
+ 16,
35
+ 17,
36
+ 18,
37
+ 19,
38
+ 20,
39
+ 21,
40
+ 22,
41
+ 23,
42
+ 24,
43
+ 25,
44
+ 26,
45
+ 27,
46
+ 28,
47
+ 29,
48
+ 30,
49
+ 31,
50
+ 32,
51
+ 33,
52
+ 34,
53
+ 35,
54
+ 36,
55
+ 37,
56
+ 38,
57
+ 39,
58
+ 40,
59
+ 41,
60
+ 42,
61
+ 43,
62
+ 44,
63
+ 45,
64
+ 46,
65
+ 47,
66
+ 48,
67
+ 49,
68
+ 50,
69
+ 51,
70
+ 52,
71
+ 53,
72
+ 54,
73
+ 55,
74
+ 56,
75
+ 57,
76
+ 58,
77
+ 59,
78
+ 60,
79
+ 61,
80
+ 62,
81
+ 63,
82
+ 64,
83
+ 65,
84
+ 66,
85
+ 67,
86
+ 68,
87
+ 69,
88
+ 70,
89
+ 71,
90
+ 72,
91
+ 73,
92
+ 74,
93
+ 75,
94
+ 76,
95
+ 77,
96
+ 78,
97
+ 79,
98
+ 80,
99
+ 81,
100
+ 82,
101
+ 83,
102
+ 84,
103
+ 85,
104
+ 86,
105
+ 87,
106
+ 88,
107
+ 89,
108
+ 90,
109
+ 91,
110
+ 92,
111
+ 93,
112
+ 94,
113
+ 95,
114
+ 96,
115
+ 97,
116
+ 98,
117
+ 99,
118
+ 100,
119
+ 101,
120
+ 102,
121
+ 103,
122
+ 104,
123
+ 105,
124
+ 106,
125
+ 107,
126
+ 108,
127
+ 109,
128
+ 110,
129
+ 111,
130
+ 112,
131
+ 113,
132
+ 114,
133
+ 115,
134
+ 116,
135
+ 117,
136
+ 118,
137
+ 119,
138
+ 120,
139
+ 121,
140
+ 122,
141
+ 123,
142
+ 124,
143
+ 125,
144
+ 126,
145
+ 127,
146
+ 128,
147
+ 129,
148
+ 130,
149
+ 131,
150
+ 132,
151
+ 133,
152
+ 134,
153
+ 135,
154
+ 136,
155
+ 137,
156
+ 138,
157
+ 139,
158
+ 140,
159
+ 141,
160
+ 142,
161
+ 143,
162
+ 144,
163
+ 145,
164
+ 146,
165
+ 147,
166
+ 148,
167
+ 149,
168
+ 150,
169
+ 151,
170
+ 152,
171
+ 153,
172
+ 154,
173
+ 155,
174
+ 156,
175
+ 157,
176
+ 158,
177
+ 159,
178
+ 160,
179
+ 161,
180
+ 162,
181
+ 163,
182
+ 164,
183
+ 165,
184
+ 166,
185
+ 167,
186
+ 168
187
+ ]
188
+ },
189
+ {
190
+ "lr": 1e-05,
191
+ "betas": [
192
+ 0.9,
193
+ 0.999
194
+ ],
195
+ "eps": 1e-08,
196
+ "weight_decay": 0.0001,
197
+ "amsgrad": false,
198
+ "maximize": false,
199
+ "foreach": null,
200
+ "capturable": false,
201
+ "differentiable": false,
202
+ "fused": null,
203
+ "decoupled_weight_decay": true,
204
+ "params": [
205
+ 169,
206
+ 170,
207
+ 171,
208
+ 172,
209
+ 173,
210
+ 174,
211
+ 175,
212
+ 176,
213
+ 177,
214
+ 178,
215
+ 179,
216
+ 180,
217
+ 181,
218
+ 182,
219
+ 183,
220
+ 184,
221
+ 185,
222
+ 186,
223
+ 187,
224
+ 188
225
+ ]
226
+ }
227
+ ]
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/training_state/optimizer_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:785cfe8fe7eb7fcad3aae9fc1059d2fd9b5fed20aa8bc542797cfc856da83880
3
+ size 514551740
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/training_state/rng_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1adee2b9b73a93dc6c1ca5e6aca00b64a17400b8df742713faf150a83a72268d
3
+ size 15708
outputs/train/zeno-ai/BookOriginRelocation/streaming-act-prism/20260504_103716/checkpoints/060000/training_state/training_step.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "step": 60000
3
+ }