yslan commited on
Commit
939af89
·
verified ·
1 Parent(s): 85d3649

Omni4D stage-1 4D model, update 15,000

Browse files
stage1_4d_wan22_5b_step_015000/README.md ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ # Omni4D stage-1 4D model, update 15,000 (2026-10-29)
2
+
3
+ Wan2.2-TI2V-5B with separate token streams: video, depth image (GenCeption log depth), RayMap camera, DINOv3 features; joint attention in blocks 7-22, stream-only blocks 0-6 and 23-29; modality forcing. Trained on the 4D mixture (Syn4D, SynthVerse, ParallelDomain-4D, OmniWorld-Game, Spring, TartanAir, vKITTI, Dynamic Replica, MidAir, MVS-Synth, TartanGround), 9 frames at 288 x 512.
4
+
5
+ - weights_step_015000_bf16.pt: inference weights (12.1 GB).
6
+ - fp32_masters_step_015000.pt: FP32 master weights, the start of the stage-1 continuation and robot post-training runs (24.2 GB); fp32_masters_step_015000.json lists the tensor groups.
7
+ - config.yaml: the run config (resume at 10,000 to 15,000).
8
+ - code_commit.txt: training code commit (repo NIRVANALAN/vgg_forecast_4d).
stage1_4d_wan22_5b_step_015000/code_commit.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ commit cdf38ca61fb27e63f68d017bf00cd5d661174e22
stage1_4d_wan22_5b_step_015000/config.yaml ADDED
@@ -0,0 +1,374 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ output_dir: /scratch/shared/engs2907/logs/vgg_forecast_4d/train/flexpi_four_d_pretraining_b200_500k_updates_relaunch_captions_depth_image_20261029/98e01ac6b/run
2
+ gpu_model: B200
3
+ batch_size: 8
4
+ num_workers: 8
5
+ learning_rate: 0.0001
6
+ num_epochs: 20
7
+ max_steps: 500000
8
+ log_every: 10
9
+ save_every: 1000
10
+ eval_every: 0
11
+ eval_num_inference_steps: 25
12
+ keep_last_n_states: 2
13
+ keep_last_n_weights: 5
14
+ eval_video: false
15
+ share_pca_basis: true
16
+ gradient_accumulation_steps: 4
17
+ mixed_precision: bf16
18
+ seed: 42
19
+ max_grad_norm: 1.0
20
+ weight_decay: 0.01
21
+ resume: /scratch/shared/engs2907/logs/vgg_forecast_4d/train/flexpi_four_d_pretraining_b200_500k_updates_20261028/4443e0637/run/checkpoints/state/step_010000
22
+ resume_reanchor_lr_schedule: false
23
+ resume_config_source: current
24
+ pretrained_ckpt: null
25
+ pretrained_ckpt_strict_shape: false
26
+ wandb:
27
+ enabled: true
28
+ workspace: null
29
+ project: omni4d-robotwin
30
+ name: flexpi_four_d_pretraining_b200_500k_relaunch_captions_depth_image
31
+ group: flexpi_four_d_pretraining
32
+ mode: online
33
+ data:
34
+ train:
35
+ _target_: four_d_windows.FourDPretrainingWindows
36
+ sources:
37
+ - name: four_d_pretraining_relaunch_rendered_depth_sources
38
+ kind: track4d_source_config
39
+ config: train/mot/wan_video_depth_vae_from_video_four_d_pretraining_relaunch_rendered_depth_sources_21_frames_512
40
+ weight: 40.5
41
+ left_out_sources:
42
+ - omniworld_game
43
+ held_out_left_out_sources: []
44
+ frames: 9
45
+ raster_hw:
46
+ - 288
47
+ - 512
48
+ raymap_camera:
49
+ origin_fraction: 0.3333333333333333
50
+ origin_range: 1.0
51
+ decode_margin: 16
52
+ origin_box_grid: 1
53
+ pointmap_log_factor: 0.4
54
+ pointmap_hole_fill:
55
+ synthverse_depth: far_clip
56
+ syn4d: far_clip
57
+ omniworld_game: far_clip
58
+ dynamic_replica: far_clip
59
+ tartanair: far_clip
60
+ dl3dv: far_clip
61
+ scannet: telea_log_depth
62
+ virtual_kitti2: far_clip
63
+ midair: far_clip
64
+ pointodyssey: far_clip
65
+ mvs_synth: far_clip
66
+ wildrgbd: telea_log_depth
67
+ hypersim: far_clip
68
+ spring: far_clip
69
+ kubric49: far_clip
70
+ kubric81: far_clip
71
+ droid_stereo_depth: telea_log_depth
72
+ pointmap_gauge: window_labelled_median_depth
73
+ geometry_image: genception_log_depth
74
+ context_len: 512
75
+ batch_size: 8
76
+ stream_seed: 42
77
+ epoch_windows: 32000000
78
+ val_set_proportion: 0.0
79
+ resume_state: /scratch/shared/engs2907/logs/vgg_forecast_4d/train/flexpi_four_d_pretraining_b200_500k_updates_20261028/4443e0637/run/checkpoints/state/step_010000
80
+ gradient_accumulation_steps: 4
81
+ loader_workers: 8
82
+ loader_opencv_threads: 1
83
+ window_captions:
84
+ caption_dropout: 0.1
85
+ missing_caption: empty_prompt
86
+ caches:
87
+ syn4d_v1_gemini_holo4d:
88
+ embeddings: /scratch/shared/engs2907/logs/vgg_forecast_4d/cache/wan_umt5_captions/four_d_window_captions_512/syn4d_v1_gemini_holo4d/captions_t5_emb.safetensors
89
+ meta: /scratch/shared/engs2907/logs/vgg_forecast_4d/cache/wan_umt5_captions/four_d_window_captions_512/syn4d_v1_gemini_holo4d/captions_t5_emb_meta.json
90
+ syn4d_v2_gemini_holo4d:
91
+ embeddings: /scratch/shared/engs2907/logs/vgg_forecast_4d/cache/wan_umt5_captions/four_d_window_captions_512/syn4d_v2_gemini_holo4d/captions_t5_emb.safetensors
92
+ meta: /scratch/shared/engs2907/logs/vgg_forecast_4d/cache/wan_umt5_captions/four_d_window_captions_512/syn4d_v2_gemini_holo4d/captions_t5_emb_meta.json
93
+ tartanair_gemini_holo4d:
94
+ embeddings: /scratch/shared/engs2907/logs/vgg_forecast_4d/cache/wan_umt5_captions/four_d_window_captions_512/tartanair_gemini_holo4d/captions_t5_emb.safetensors
95
+ meta: /scratch/shared/engs2907/logs/vgg_forecast_4d/cache/wan_umt5_captions/four_d_window_captions_512/tartanair_gemini_holo4d/captions_t5_emb_meta.json
96
+ mvs_synth_gemini_holo4d:
97
+ embeddings: /scratch/shared/engs2907/logs/vgg_forecast_4d/cache/wan_umt5_captions/four_d_window_captions_512/mvs_synth_gemini_holo4d/captions_t5_emb.safetensors
98
+ meta: /scratch/shared/engs2907/logs/vgg_forecast_4d/cache/wan_umt5_captions/four_d_window_captions_512/mvs_synth_gemini_holo4d/captions_t5_emb_meta.json
99
+ stanzas:
100
+ syn4d_v1:
101
+ product: syn4d_v1_gemini_holo4d
102
+ join: global
103
+ syn4d_v2_mixed:
104
+ product: syn4d_v2_gemini_holo4d
105
+ join: local_covering
106
+ syn4d_v2_mixed_no_sim:
107
+ product: syn4d_v2_gemini_holo4d
108
+ join: local_covering
109
+ tartanair:
110
+ product: tartanair_gemini_holo4d
111
+ join: local_covering
112
+ mvs_synth:
113
+ product: mvs_synth_gemini_holo4d
114
+ join: local_covering
115
+ omniworld_game: none
116
+ pointodyssey_test: none
117
+ kubric49_validation: none
118
+ wildrgbd_test: none
119
+ vkitti_test: none
120
+ droid_held_out: none
121
+ model:
122
+ _target_: flexpi_track_video.create_flexpi_track_video
123
+ model_id: Wan-AI/Wan2.2-TI2V-5B
124
+ tokenizer_model_id: Wan-AI/Wan2.1-T2V-1.3B
125
+ tokenizer_max_len: 128
126
+ load_text_encoder: false
127
+ proprio_dim: 14
128
+ redirect_common_files: true
129
+ mot_checkpoint_mixed_attn: true
130
+ action_dit_pretrained_path: /scratch/shared/shug6902/logs/vgg_forecast_4d/cache/flexpi_robotwin_native_training_20261019/models/ActionDiT_linear_interp_Wan22_alphascale_1024hdim.pt
131
+ skip_dit_load_from_pretrain: false
132
+ dino_dim: 768
133
+ dino_model_name: vit_base_patch16_dinov3.lvd1689m
134
+ composite_layout: tshape_robotwin_384x320_uniform_dino_on_video_grid
135
+ dino_cam_patches: null
136
+ dino_cam_regions: null
137
+ dino_temporal_stride: 1
138
+ dino_stride_keep_far: true
139
+ dino_pred_x0: true
140
+ dino_pool_mode: avg
141
+ freeze_dino_encoder: true
142
+ dino_pixel_unshuffle: 2
143
+ enable_pointmap: true
144
+ pointmap_max_depth_m: 2.0
145
+ pointmap_norm_bounds:
146
+ x:
147
+ - -0.5
148
+ - 0.5
149
+ 'y':
150
+ - -0.5
151
+ - 0.5
152
+ z:
153
+ - 0.0
154
+ - 1.5
155
+ pointmap_depth_vis_mode: turbo
156
+ joint_video: true
157
+ joint_dino: true
158
+ joint_pointmap: true
159
+ flex_joint:
160
+ enabled: true
161
+ p_present_video: 0.5
162
+ p_present_dino: 0.5
163
+ p_present_pointmap: 0.5
164
+ p_jv: 0.5
165
+ p_jd: 0.5
166
+ p_jp: 0.5
167
+ cross_modal_predict_video: true
168
+ cross_modal_predict_dino: true
169
+ cross_modal_predict_pointmap: true
170
+ video_dit_config:
171
+ has_image_input: false
172
+ patch_size:
173
+ - 1
174
+ - 2
175
+ - 2
176
+ in_dim: 48
177
+ hidden_dim: 3072
178
+ ffn_dim: 14336
179
+ freq_dim: 256
180
+ text_dim: 4096
181
+ out_dim: 48
182
+ num_heads: 24
183
+ attn_head_dim: 128
184
+ num_layers: 30
185
+ eps: 1.0e-06
186
+ seperated_timestep: true
187
+ require_clip_embedding: false
188
+ require_vae_embedding: false
189
+ fuse_vae_embedding_in_latents: true
190
+ use_gradient_checkpointing: true
191
+ video_attention_mask_mode: first_frame_causal
192
+ action_conditioned: false
193
+ action_dim: 14
194
+ action_group_causal_mask_mode: group_diagonal
195
+ action_dit_config:
196
+ action_dim: 14
197
+ hidden_dim: 1024
198
+ ffn_dim: 4096
199
+ num_heads: 24
200
+ attn_head_dim: 128
201
+ num_layers: 30
202
+ text_dim: 4096
203
+ freq_dim: 256
204
+ eps: 1.0e-06
205
+ use_gradient_checkpointing: true
206
+ video_scheduler:
207
+ train_shift: 6.0
208
+ infer_shift: 6.0
209
+ num_train_timesteps: 1000
210
+ action_scheduler:
211
+ train_shift: 1.0
212
+ infer_shift: 1.0
213
+ num_train_timesteps: 1000
214
+ dino_scheduler:
215
+ train_shift: 6.0
216
+ infer_shift: 6.0
217
+ num_train_timesteps: 1000
218
+ pointmap_scheduler:
219
+ train_shift: 6.0
220
+ infer_shift: 6.0
221
+ num_train_timesteps: 1000
222
+ loss:
223
+ lambda_action: 1.0
224
+ lambda_dino: 1.0
225
+ lambda_pointmap: 1.0
226
+ hbridge:
227
+ enabled: true
228
+ bottom_ratio: 0.25
229
+ top_ratio: 0.25
230
+ track_video:
231
+ attention: visual_stream
232
+ sparse_action: true
233
+ train_shift: 6.0
234
+ infer_shift: 6.0
235
+ loss_lambda: 1.0
236
+ action_edge: two_way
237
+ flex:
238
+ p_present: 1.0
239
+ p_joint: 0.5
240
+ cross_modal_predict: true
241
+ noise_level_coupling: independent
242
+ raymap_camera:
243
+ origin_fraction: 0.3333333333333333
244
+ origin_range: 1.0
245
+ decode_margin: 16
246
+ origin_box_grid: 1
247
+ train_shift: 6.0
248
+ infer_shift: 6.0
249
+ loss_lambda: 1.0
250
+ four_d_pretraining:
251
+ geometry_streams: generated
252
+ dino_stream: generated
253
+ video_loss_lambda: 1.0
254
+ modality_forcing:
255
+ video:
256
+ p_present: 0.5
257
+ p_joint: 0.5
258
+ cross_modal: true
259
+ dino:
260
+ p_present: 0.5
261
+ p_joint: 0.5
262
+ cross_modal: true
263
+ pointmap:
264
+ p_present: 0.5
265
+ p_joint: 0.5
266
+ cross_modal: true
267
+ camera:
268
+ p_present: 0.5
269
+ p_joint: 0.5
270
+ cross_modal: true
271
+ training_mode_shares:
272
+ flexible_joint: 0.6
273
+ v2d: 0.04
274
+ v2c: 0.04
275
+ c2v_from_first_frame: 0.032
276
+ c2vd_from_first_frame: 0.032
277
+ c2v: 0.008
278
+ c2vd: 0.008
279
+ vd2c: 0.025
280
+ vc2d: 0.025
281
+ v2dc: 0.06
282
+ dc2v_from_first_frame: 0.02
283
+ d2vc_from_first_frame: 0.02
284
+ d2v_from_first_frame: 0.02
285
+ dc2v: 0.005
286
+ d2vc: 0.005
287
+ d2v: 0.005
288
+ d2c: 0.0125
289
+ c2d: 0.0125
290
+ v2d_dino_generated: 0.01
291
+ v2c_dino_generated: 0.01
292
+ v2dc_dino_generated: 0.01
293
+ pointmap_log_factor: 0.4
294
+ minimum_learning_rate: 1.0e-06
295
+ training_scope: four_d_pretraining_8000_updates' model and modality forcing on the
296
+ stage-1 mixture for 500,000 updates of 64 windows (8 per GPU x 2 B200 x accumulation
297
+ 4), the stage-1 schedule (warmup_steps, cosine to 1e-6, clipping at max_grad_norm),
298
+ a full state every 1,000 updates and the FP32 masters of every 10,000th
299
+ native_control_fps: 50
300
+ action_dimension: 14
301
+ initialization: native_wan_actiondit
302
+ track_video_encoding:
303
+ record_root: /scratch/shared/engs2907/logs/vgg_forecast_4d/data/robotwin_robot_records_all_tasks_7bf528a2_20261023
304
+ embodiment_root: /scratch/shared/engs2907/checkouts/robotwin_objects_b5230d8b_266f3aad/RoboTwin/assets/embodiments/aloha-agilex
305
+ surface_points: 8192
306
+ min_surface_points_per_body: 16
307
+ query_count:
308
+ - 64
309
+ - 256
310
+ camera_query_quota:
311
+ - 224
312
+ - 16
313
+ - 16
314
+ min_visible_queries: 16
315
+ query_draw: per_read
316
+ query_seed: 0
317
+ visibility_margin_m: 0.01
318
+ pixel_center: 0.0
319
+ colouring: frame0_head_xyz
320
+ xyz_normalization: quantile_0.02_0.98
321
+ hues: null
322
+ neighbour_px: null
323
+ colour_floor: 0.2
324
+ min_gap_px:
325
+ - 8
326
+ - 6
327
+ - 6
328
+ wrist_dense_links:
329
+ - fl_link7
330
+ - fl_link8
331
+ - fr_link7
332
+ - fr_link8
333
+ wrist_dense_points: 16384
334
+ disc_radius_px: 2
335
+ occluded_style: ring
336
+ ring_radius_px: 3
337
+ ring_width_px: 1
338
+ deepspeed_gradient_clipping: true
339
+ warmup_steps: 1000
340
+ loader_prefetch_factor: 2
341
+ loader_opencv_threads: 1
342
+ weights_milestone_every: 10000
343
+ fp32_milestone_every: 10000
344
+ data_segment:
345
+ name: four_d_relaunch_captions_and_genception_depth_image
346
+ review: doc/review/2026-10-29_four_d_pretraining_window_captions.md
347
+ changed_keys:
348
+ - output_dir
349
+ - wandb.name
350
+ - data.train.sources
351
+ - data.train.pointmap_gauge
352
+ - data.train.geometry_image
353
+ - data.train.context_len
354
+ - data.train.window_captions
355
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.v2d
356
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.v2c
357
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.c2v_from_first_frame
358
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.c2vd_from_first_frame
359
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.c2v
360
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.c2vd
361
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.vd2c
362
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.vc2d
363
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.v2dc
364
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.dc2v_from_first_frame
365
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.d2vc_from_first_frame
366
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.d2v_from_first_frame
367
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.dc2v
368
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.d2vc
369
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.d2v
370
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.d2c
371
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.c2d
372
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.v2d_dino_generated
373
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.v2c_dino_generated
374
+ - model.four_d_pretraining.modality_forcing.training_mode_shares.v2dc_dino_generated
stage1_4d_wan22_5b_step_015000/fp32_masters_step_015000.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "state_dir": "/scratch/shared/engs2907/logs/vgg_forecast_4d/train/flexpi_four_d_pretraining_b200_500k_updates_relaunch_captions_depth_image_20261029/98e01ac6b/run/checkpoints/state/step_015000",
3
+ "deepspeed_tag": "pytorch_model",
4
+ "weights_layout": "/scratch/shared/engs2907/logs/vgg_forecast_4d/train/flexpi_four_d_pretraining_b200_500k_updates_relaunch_captions_depth_image_20261029/98e01ac6b/run/checkpoints/weights/step_015000.pt",
5
+ "step": 15000,
6
+ "torch_dtype": "torch.float32",
7
+ "groups": {
8
+ "mot": {
9
+ "tensors": 1649,
10
+ "elements": 6020688078
11
+ },
12
+ "dino_embedder": {
13
+ "tensors": 2,
14
+ "elements": 9440256
15
+ },
16
+ "dino_proj_out": {
17
+ "tensors": 2,
18
+ "elements": 9440256
19
+ },
20
+ "dino_feature_norm": {
21
+ "tensors": 2,
22
+ "elements": 6144
23
+ },
24
+ "pt_patch_embedding": {
25
+ "tensors": 2,
26
+ "elements": 592896
27
+ },
28
+ "pt_head": {
29
+ "tensors": 3,
30
+ "elements": 596160
31
+ },
32
+ "av_patch_embedding": {
33
+ "tensors": 2,
34
+ "elements": 592896
35
+ },
36
+ "av_head": {
37
+ "tensors": 3,
38
+ "elements": 596160
39
+ },
40
+ "camera_stream": {
41
+ "tensors": 5,
42
+ "elements": 1189056
43
+ },
44
+ "proprio_encoder": {
45
+ "tensors": 2,
46
+ "elements": 61440
47
+ }
48
+ },
49
+ "expert_masters_named": {
50
+ "video": [
51
+ "video_expert.",
52
+ "mot.mixtures.video."
53
+ ],
54
+ "action": [
55
+ "action_expert.",
56
+ "mot.mixtures.action."
57
+ ]
58
+ },
59
+ "masters": 3321,
60
+ "masters_left_out_by_module": {},
61
+ "out": "/scratch/shared/engs2907/logs/vgg_forecast_4d/train/flexpi_four_d_pretraining_b200_500k_updates_relaunch_captions_depth_image_20261029/98e01ac6b/run/checkpoints/fp32/step_015000.pt"
62
+ }
stage1_4d_wan22_5b_step_015000/fp32_masters_step_015000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bdc3fe17dd1ad0e554f81ecad94820ba24ae823be19d859a0f46264b19d0dca0
3
+ size 24173372807
stage1_4d_wan22_5b_step_015000/fp32_masters_step_015000.verified.json ADDED
@@ -0,0 +1,126 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "step": 15000,
3
+ "verified": true,
4
+ "run_commit": "98e01ac6bbd0f8153381c8893cc91fed3f76b3d4",
5
+ "consolidation_commit": "cdf38ca61fb27e63f68d017bf00cd5d661174e22",
6
+ "fp32_file": {
7
+ "path": "/scratch/shared/engs2907/logs/vgg_forecast_4d/train/flexpi_four_d_pretraining_b200_500k_updates_relaunch_captions_depth_image_20261029/98e01ac6b/run/checkpoints/fp32/step_015000.pt",
8
+ "bytes": 24173372807,
9
+ "sha256": "bdc3fe17dd1ad0e554f81ecad94820ba24ae823be19d859a0f46264b19d0dca0",
10
+ "torch_dtype": "torch.float32",
11
+ "reloaded_equal": true
12
+ },
13
+ "source_state": {
14
+ "path": "/scratch/shared/engs2907/logs/vgg_forecast_4d/train/flexpi_four_d_pretraining_b200_500k_updates_relaunch_captions_depth_image_20261029/98e01ac6b/run/checkpoints/state/step_015000",
15
+ "files": {
16
+ "latest": {
17
+ "bytes": 13,
18
+ "sha256": "68d0ca762047fd48fdd6a06a5e6465209289c028bfd63f5557a110ff1ebc15ac"
19
+ },
20
+ "pytorch_model/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt": {
21
+ "bytes": 36259264113,
22
+ "sha256": "6bec0361a08aa7972407f94e35126a544e1895d03403df1c7ce4b689ad8af084"
23
+ },
24
+ "pytorch_model/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt": {
25
+ "bytes": 36259320177,
26
+ "sha256": "43f7189acf78e8b88eac2def4ee998e48c737b22b0a222ad07b8330523b1622b"
27
+ },
28
+ "pytorch_model/mp_rank_00_model_states.pt": {
29
+ "bytes": 13668502141,
30
+ "sha256": "4c1157ac1ea46933b7448a722c8aa97024871bf64b33f0d4a1e0e2cf3c1ada87"
31
+ },
32
+ "random_states_0.pkl": {
33
+ "bytes": 15017,
34
+ "sha256": "7c52045b70f89d865150b1cdae6d6cd9747fd282bcb7c668ca70fb150f6b4b3d"
35
+ },
36
+ "random_states_1.pkl": {
37
+ "bytes": 15017,
38
+ "sha256": "7819b5d724532d61b7f8e37e68d5a68e10f8ed23c1dc2a531eb35c33fdb0c479"
39
+ },
40
+ "scheduler.bin": {
41
+ "bytes": 1593,
42
+ "sha256": "b47a9f40efac2fed081c57733261fabfa0412494ec04de6b5b834c6acce87726"
43
+ },
44
+ "trainer_state.json": {
45
+ "bytes": 67,
46
+ "sha256": "3c03531b71b6a60272282488953fd865ab116c0fb9a1259a09f19e95dc60116a"
47
+ },
48
+ "zero_to_fp32.py": {
49
+ "bytes": 33270,
50
+ "sha256": "e8a2553b434ac043ec63b022db1e768277acf8d4e70f10a2842db6d238977135"
51
+ }
52
+ },
53
+ "manifest_sha256": "a719a57a8e03ed0606c3dca81fe817cca218e797fd25b4ff419d1228d4f4692b"
54
+ },
55
+ "weights_file": {
56
+ "path": "/scratch/shared/engs2907/logs/vgg_forecast_4d/train/flexpi_four_d_pretraining_b200_500k_updates_relaunch_captions_depth_image_20261029/98e01ac6b/run/checkpoints/weights/step_015000.pt",
57
+ "bytes": 12086724089,
58
+ "sha256": "a06a4549dddf549d95cc24260734af81b8ccecf3e0d5a25e4bacfdc523c4d390"
59
+ },
60
+ "mapping": {
61
+ "groups": {
62
+ "mot": {
63
+ "tensors": 1649,
64
+ "elements": 6020688078
65
+ },
66
+ "dino_embedder": {
67
+ "tensors": 2,
68
+ "elements": 9440256
69
+ },
70
+ "dino_proj_out": {
71
+ "tensors": 2,
72
+ "elements": 9440256
73
+ },
74
+ "dino_feature_norm": {
75
+ "tensors": 2,
76
+ "elements": 6144
77
+ },
78
+ "pt_patch_embedding": {
79
+ "tensors": 2,
80
+ "elements": 592896
81
+ },
82
+ "pt_head": {
83
+ "tensors": 3,
84
+ "elements": 596160
85
+ },
86
+ "av_patch_embedding": {
87
+ "tensors": 2,
88
+ "elements": 592896
89
+ },
90
+ "av_head": {
91
+ "tensors": 3,
92
+ "elements": 596160
93
+ },
94
+ "camera_stream": {
95
+ "tensors": 5,
96
+ "elements": 1189056
97
+ },
98
+ "proprio_encoder": {
99
+ "tensors": 2,
100
+ "elements": 61440
101
+ }
102
+ },
103
+ "layout_keys": 1672,
104
+ "every_layout_key_has_a_master": true,
105
+ "masters_round_to_the_weights_file": true,
106
+ "masters_left_out_by_module": {}
107
+ },
108
+ "codec": {
109
+ "pointmap": {
110
+ "normalization": "genception_log",
111
+ "log_factor": 0.4,
112
+ "gauge": "window_labelled_median_depth",
113
+ "gauge_frame": "every_frame",
114
+ "geometry_image": "genception_log_depth",
115
+ "validity": "finite_and_positive_z"
116
+ },
117
+ "raymap_camera": {
118
+ "origin_fraction": 0.3333333333333333,
119
+ "origin_range": 1.0,
120
+ "decode_margin": 16,
121
+ "origin_box_grid": 1,
122
+ "translation": "window_median_depth_scale"
123
+ },
124
+ "dino_positions": "video_grid"
125
+ }
126
+ }
stage1_4d_wan22_5b_step_015000/weights_step_015000_bf16.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a06a4549dddf549d95cc24260734af81b8ccecf3e0d5a25e4bacfdc523c4d390
3
+ size 12086724089