Upload folder using huggingface_hub

.gitattributes

@@ -544,3 +544,18 @@ bimamba_stage_b_joint_1024_40k_fixed/wandb/offline-run-20260420_034010-stage_b_j
 bimamba_stage_b_joint_1024_40k_fixed/wandb/offline-run-20260421_131537-stage_b_joint_offline/files/media/videos/validation_vis/video_0_0_43e937b607e356d95a84.mp4 filter=lfs diff=lfs merge=lfs -text
 bimamba_stage_b_joint_1024_40k_fixed/wandb/offline-run-20260421_131537-stage_b_joint_offline/files/media/videos/validation_vis/video_0_102_917f987a7e7c2f3cdedf.mp4 filter=lfs diff=lfs merge=lfs -text
 bimamba_stage_b_joint_1024_40k_fixed/wandb/offline-run-20260421_131537-stage_b_joint_offline/run-stage_b_joint_offline.wandb filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_1020_4b9fdaf08894b40ece9f.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_1122_36c82a57b0ef7b0bd013.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_1224_6b7779c9e126fa9b302f.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_714_8c2a277230d1b61fa63d.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_816_9528e9623fc2d0e7abfa.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_918_f8392fb64b9a16233691.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_0_d99c30e3f713e6884745.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_102_55531416f1a0f46778e0.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_204_b417c4bcea7d8fabb351.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_306_90b4cff004dbef7461ac.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_408_5df38886c179815b23f2.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_510_fcf8664bcf56b448556a.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_612_ffb9a55464b8c069ef8b.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_714_ac56a01406adfbb1f0de.mp4 filter=lfs diff=lfs merge=lfs -text
+hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/run-stage_b_joint_offline.wandb filter=lfs diff=lfs merge=lfs -text

hierarchy_bimamba_stage_b_joint_fixed/checkpoints/epoch0_step60000.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:59d699b6a4978e6da2e5f3371b631ec0dec369126147601a82a242f7e1072a36
+size 9674914923

hierarchy_bimamba_stage_b_joint_fixed/checkpoints/epoch0_step77500.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec0f5b35381f9963a1d40a1fec1975a80b423a5141b9fec5105fc4f12a78ab0d
+size 9674914987

hierarchy_bimamba_stage_b_joint_fixed/videos/validation_vis/gt/video_0_rank0_step0.mp4

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a65a9d93cf192d8fd999e3e851b6314a14a960f11529a153c1d43a809b12dae1
-size 4013025
+oid sha256:ac49282d9417e32c6d6c8db523b59f87883fa5b920aba1857e20e8ea4be232bb
+size 5123924

hierarchy_bimamba_stage_b_joint_fixed/videos/validation_vis/pred/video_0_rank0_step0.mp4

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f91d7ff4850ebaeeecfda00f41e506ea215bd6bc8c8cea4fa97fff8a5a107a94
-size 5717157
+oid sha256:97b42e15f339da8ab53075cf6b7214d5f662830c491581a7fb8fcc98f7a55fe1
+size 4758055

hierarchy_bimamba_stage_b_joint_fixed/wandb/debug.log

@@ -1,25 +1,26 @@
-2026-04-19 21:09:28,496 INFO    MainThread:124537 [wandb_setup.py:_flush():77] Current SDK version is 0.17.9
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_setup.py:_flush():77] Configure stats pid to 124537
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_setup.py:_flush():77] Loading settings from /home/x_fahkh/.config/wandb/settings
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_setup.py:_flush():77] Loading settings from /proj/cvl/users/x_fahkh2/WorldMem_Repro/wandb/settings
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_setup.py:_flush():77] Loading settings from environment variables: {'disabled': 'true'}
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_setup.py:_flush():77] Applying setup settings: {'_disable_service': False}
-2026-04-19 21:09:28,497 WARNING MainThread:124537 [wandb_setup.py:_flush():77] Could not find program at -m main
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_setup.py:_flush():77] Inferring run settings from compute environment: {'program_relpath': None, 'program': '-m main'}
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_init.py:_log_setup():524] Logging user logs to /proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/logs/debug.log
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_init.py:_log_setup():525] Logging internal logs to /proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/logs/debug-internal.log
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_init.py:init():608] calling init triggers
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_init.py:init():615] wandb.init called with sweep_config: {}
-config: {'experiment': {'debug': '${debug}', 'tasks': ['training'], 'num_nodes': 1, 'training': {'precision': '16-mixed', 'compile': False, 'lr': 2e-05, 'batch_size': 8, 'max_epochs': -1, 'max_steps': 60000, 'max_time': None, 'data': {'num_workers': 4, 'shuffle': True}, 'optim': {'accumulate_grad_batches': 1, 'gradient_clip_val': 1.0}, 'checkpointing': {'every_n_train_steps': 2500, 'every_n_epochs': None, 'train_time_interval': None, 'enable_version_counter': False}}, 'validation': {'precision': '16-mixed', 'compile': False, 'batch_size': 4, 'val_every_n_step': 2500, 'val_every_n_epoch': None, 'limit_batch': 1, 'inference_mode': False, 'data': {'num_workers': 4, 'shuffle': False}}, 'test': {'precision': '16-mixed', 'compile': False, 'batch_size': 1, 'limit_batch': 1, 'inference_mode': False, 'data': {'num_workers': 4, 'shuffle': False}}, 'logging': {'metrics': None}, '_name': 'exp_video'}, 'dataset': {'debug': '${debug}', 'metadata': 'data/${dataset.name}/metadata.json', 'data_mean': 0.5, 'data_std': 0.5, 'save_dir': '/proj/cvl/users/x_fahkh2/WorldMem_Repro/datasets/minecraft', 'n_frames': 200, 'context_length': 1, 'resolution': 128, 'observation_shape': [3, '${dataset.resolution}', '${dataset.resolution}'], 'external_cond_dim': 0, 'validation_multiplier': 1, 'frame_skip': 1, 'action_cond_dim': 25, '_name': 'video_minecraft', 'n_frames_valid': 200, 'angle_range': 110, 'pos_range': 8, 'wo_updown': False, 'customized_validation': True, 'add_timestamp_embedding': True, 'use_explicit_memory_frames': False}, 'algorithm': {'debug': '${debug}', 'lr': '${experiment.training.lr}', 'x_shape': '${dataset.observation_shape}', 'frame_stack': 1, 'frame_skip': '${dataset.frame_skip}', 'data_mean': '${dataset.data_mean}', 'data_std': '${dataset.data_std}', 'external_cond_dim': 0, 'context_frames': 100, 'weight_decay': 0.002, 'warmup_steps': 1000, 'optimizer_beta': [0.9, 0.99], 'uncertainty_scale': 1, 'guidance_scale': 0.0, 'chunk_size': 1, 'scheduling_matrix': 'autoregressive', 'noise_level': 'random_all', 'causal': True, 'diffusion': {'objective': 'pred_v', 'beta_schedule': 'sigmoid', 'schedule_fn_kwargs': {}, 'clip_noise': 20.0, 'use_snr': False, 'use_cum_snr': False, 'use_fused_snr': True, 'snr_clip': 5.0, 'cum_snr_decay': 0.96, 'timesteps': 1000, 'sampling_timesteps': 20, 'ddim_sampling_eta': 0.0, 'stabilization_level': 15, 'architecture': {'network_size': 64, 'attn_heads': 4, 'attn_dim_head': 64, 'dim_mults': [1, 2, 4, 8], 'resolution': '${dataset.resolution}', 'attn_resolutions': [16, 32, 64, 128], 'use_init_temporal_attn': True, 'use_linear_attn': True, 'time_emb_type': 'rotary'}}, 'n_frames': '${dataset.n_frames}', 'metadata': '${dataset.metadata}', 'action_cond_dim': 25, 'use_plucker': True, 'memory_condition_length': 0, 'log_video': True, 'use_mamba_memory_pipeline': True, 'training_stage': 'stage_b_diffusion_training', 'stage_b_joint_training': True, 'stage_b_memory_aux_weight': 0.1, 'stage_b_prefix_corruption_prob': 0.0, 'stage_b_prefix_corruption_timestep_range': [5, 50], 'stage_b_prefix_corruption_max_recent_frames': 32, 'diff_window_size': 8, 'use_precomputed_features': False, 'mamba_latent_channels': 16, 'mamba_model_dim': 256, 'mamba_depth': 4, 'mamba_cond_dim': 256, 'mamba_d_state': 16, 'mamba_d_conv': 4, 'mamba_expand': 2, 'memory_scene_grid': [2, 4], 'memory_num_tokens': 4, 'memory_retrieval_topk': 32, 'allow_mamba_fallback': False, 'strict_causal_training': True, 'strict_causal_evaluation': True, 'use_oracle_pose_eval': True, 'enable_memory_noise_curriculum': False, 'curriculum_phase_boundaries': [0.2, 0.7], 'curriculum_noise_ranges': [[600, 1000], [200, 900], [0, 400]], 'curriculum_horizons': [50, 100, 200], '_name': 'df_video_mamba3stage', 'use_memory_attention': False, 'relative_embedding': False, 'n_tokens': 8}, 'debug': False, 'wandb': {'entity': 'turlin', 'project': 'worldmem', 'mode': 'online'}, 'resume': 'stage_b_joint_offline', 'load': None, 'name': 'train_stage_b_hierarchy_bimamba_joint', 'output_dir': '/proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/'}
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_init.py:init():658] starting backend
-2026-04-19 21:09:28,497 INFO    MainThread:124537 [wandb_init.py:init():662] setting up manager
-2026-04-19 21:09:28,524 INFO    MainThread:124537 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
-2026-04-19 21:09:28,531 INFO    MainThread:124537 [wandb_init.py:init():670] backend started and connected
-2026-04-19 21:09:28,574 INFO    MainThread:124537 [wandb_init.py:init():768] updated telemetry
-2026-04-19 21:09:28,634 INFO    MainThread:124537 [wandb_init.py:init():801] communicating run to backend with 90.0 second timeout
-2026-04-19 21:09:28,638 INFO    MainThread:124537 [wandb_init.py:init():852] starting run threads in backend
-2026-04-19 21:09:35,866 INFO    MainThread:124537 [wandb_run.py:_console_start():2465] atexit reg
-2026-04-19 21:09:35,867 INFO    MainThread:124537 [wandb_run.py:_redirect():2311] redirect: wrap_raw
-2026-04-19 21:09:35,867 INFO    MainThread:124537 [wandb_run.py:_redirect():2376] Wrapping output streams.
-2026-04-19 21:09:35,867 INFO    MainThread:124537 [wandb_run.py:_redirect():2401] Redirects installed.
-2026-04-19 21:09:35,878 INFO    MainThread:124537 [wandb_init.py:init():895] run started, returning control to user process
+2026-04-21 00:35:48,797 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Current SDK version is 0.17.9
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Configure stats pid to 1288079
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Loading settings from /home/x_fahkh/.config/wandb/settings
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Loading settings from /proj/cvl/users/x_fahkh2/WorldMem_Repro/wandb/settings
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Loading settings from environment variables: {'disabled': 'true'}
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Applying setup settings: {'_disable_service': False}
+2026-04-21 00:35:48,798 WARNING MainThread:1288079 [wandb_setup.py:_flush():77] Could not find program at -m main
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Inferring run settings from compute environment: {'program_relpath': None, 'program': '-m main'}
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:_log_setup():524] Logging user logs to /proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/logs/debug.log
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:_log_setup():525] Logging internal logs to /proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/logs/debug-internal.log
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:init():608] calling init triggers
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:init():615] wandb.init called with sweep_config: {}
+config: {'experiment': {'debug': '${debug}', 'tasks': ['training'], 'num_nodes': 1, 'training': {'precision': '16-mixed', 'compile': False, 'lr': 2e-05, 'batch_size': 8, 'max_epochs': -1, 'max_steps': 175000, 'max_time': None, 'data': {'num_workers': 4, 'shuffle': True}, 'optim': {'accumulate_grad_batches': 1, 'gradient_clip_val': 1.0}, 'checkpointing': {'every_n_train_steps': 2500, 'every_n_epochs': None, 'train_time_interval': None, 'enable_version_counter': False}}, 'validation': {'precision': '16-mixed', 'compile': False, 'batch_size': 4, 'val_every_n_step': 2500, 'val_every_n_epoch': None, 'limit_batch': 1, 'inference_mode': False, 'data': {'num_workers': 4, 'shuffle': False}}, 'test': {'precision': '16-mixed', 'compile': False, 'batch_size': 1, 'limit_batch': 1, 'inference_mode': False, 'data': {'num_workers': 4, 'shuffle': False}}, 'logging': {'metrics': None}, '_name': 'exp_video'}, 'dataset': {'debug': '${debug}', 'metadata': 'data/${dataset.name}/metadata.json', 'data_mean': 0.5, 'data_std': 0.5, 'save_dir': '/proj/cvl/users/x_fahkh2/WorldMem_Repro/datasets/minecraft', 'n_frames': 200, 'context_length': 1, 'resolution': 128, 'observation_shape': [3, '${dataset.resolution}', '${dataset.resolution}'], 'external_cond_dim': 0, 'validation_multiplier': 1, 'frame_skip': 1, 'action_cond_dim': 25, '_name': 'video_minecraft', 'n_frames_valid': 200, 'angle_range': 110, 'pos_range': 8, 'wo_updown': False, 'customized_validation': True, 'add_timestamp_embedding': True, 'use_explicit_memory_frames': False}, 'algorithm': {'debug': '${debug}', 'lr': '${experiment.training.lr}', 'x_shape': '${dataset.observation_shape}', 'frame_stack': 1, 'frame_skip': '${dataset.frame_skip}', 'data_mean': '${dataset.data_mean}', 'data_std': '${dataset.data_std}', 'external_cond_dim': 0, 'context_frames': 100, 'weight_decay': 0.002, 'warmup_steps': 1000, 'optimizer_beta': [0.9, 0.99], 'uncertainty_scale': 1, 'guidance_scale': 0.0, 'chunk_size': 1, 'scheduling_matrix': 'autoregressive', 'noise_level': 'random_all', 'causal': True, 'diffusion': {'objective': 'pred_v', 'beta_schedule': 'sigmoid', 'schedule_fn_kwargs': {}, 'clip_noise': 20.0, 'use_snr': False, 'use_cum_snr': False, 'use_fused_snr': True, 'snr_clip': 5.0, 'cum_snr_decay': 0.96, 'timesteps': 1000, 'sampling_timesteps': 20, 'ddim_sampling_eta': 0.0, 'stabilization_level': 15, 'architecture': {'network_size': 64, 'attn_heads': 4, 'attn_dim_head': 64, 'dim_mults': [1, 2, 4, 8], 'resolution': '${dataset.resolution}', 'attn_resolutions': [16, 32, 64, 128], 'use_init_temporal_attn': True, 'use_linear_attn': True, 'time_emb_type': 'rotary'}}, 'n_frames': '${dataset.n_frames}', 'metadata': '${dataset.metadata}', 'action_cond_dim': 25, 'use_plucker': True, 'memory_condition_length': 0, 'log_video': True, 'use_mamba_memory_pipeline': True, 'training_stage': 'stage_b_diffusion_training', 'stage_b_joint_training': True, 'stage_b_memory_aux_weight': 0.1, 'stage_b_prefix_corruption_prob': 0.0, 'stage_b_prefix_corruption_timestep_range': [5, 50], 'stage_b_prefix_corruption_max_recent_frames': 32, 'diff_window_size': 8, 'use_precomputed_features': False, 'mamba_latent_channels': 16, 'mamba_model_dim': 256, 'mamba_depth': 4, 'mamba_cond_dim': 256, 'mamba_d_state': 16, 'mamba_d_conv': 4, 'mamba_expand': 2, 'memory_scene_grid': [2, 4], 'memory_num_tokens': 4, 'memory_retrieval_topk': 32, 'allow_mamba_fallback': False, 'strict_causal_training': True, 'strict_causal_evaluation': True, 'use_oracle_pose_eval': True, 'enable_memory_noise_curriculum': False, 'curriculum_phase_boundaries': [0.2, 0.7], 'curriculum_noise_ranges': [[600, 1000], [200, 900], [0, 400]], 'curriculum_horizons': [50, 100, 200], '_name': 'df_video_mamba3stage', 'use_memory_attention': False, 'relative_embedding': False, 'n_tokens': 8}, 'debug': False, 'wandb': {'entity': 'turlin', 'project': 'worldmem', 'mode': 'online'}, 'resume': 'stage_b_joint_offline', 'load': None, 'name': 'train_stage_b_hierarchy_bimamba_joint', 'output_dir': '/proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/'}
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:init():658] starting backend
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:init():662] setting up manager
+2026-04-21 00:35:48,813 INFO    MainThread:1288079 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
+2026-04-21 00:35:48,820 INFO    MainThread:1288079 [wandb_init.py:init():670] backend started and connected
+2026-04-21 00:35:49,492 INFO    MainThread:1288079 [wandb_init.py:init():768] updated telemetry
+2026-04-21 00:35:49,553 INFO    MainThread:1288079 [wandb_init.py:init():801] communicating run to backend with 90.0 second timeout
+2026-04-21 00:35:49,557 INFO    MainThread:1288079 [wandb_init.py:init():852] starting run threads in backend
+2026-04-21 00:35:58,257 INFO    MainThread:1288079 [wandb_run.py:_console_start():2465] atexit reg
+2026-04-21 00:35:58,257 INFO    MainThread:1288079 [wandb_run.py:_redirect():2311] redirect: wrap_raw
+2026-04-21 00:35:58,257 INFO    MainThread:1288079 [wandb_run.py:_redirect():2376] Wrapping output streams.
+2026-04-21 00:35:58,257 INFO    MainThread:1288079 [wandb_run.py:_redirect():2401] Redirects installed.
+2026-04-21 00:35:58,279 INFO    MainThread:1288079 [wandb_init.py:init():895] run started, returning control to user process
+2026-04-21 17:55:51,088 WARNING MsgRouterThr:1288079 [router.py:message_loop():77] message_loop has been closed

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_1020_4b9fdaf08894b40ece9f.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b9fdaf08894b40ece9fba63d88d70193f487e3b3bd1c79ed86fb107f3e9010a
+size 2201135

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_1122_36c82a57b0ef7b0bd013.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:36c82a57b0ef7b0bd013ef4ea47a45d2e938f5a703f23eb72ef64519a077232f
+size 2173347

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_1224_6b7779c9e126fa9b302f.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6b7779c9e126fa9b302f1161c7ee64654d660a27ee6c95c18b63f2fc8b85120d
+size 2307528

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_714_8c2a277230d1b61fa63d.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8c2a277230d1b61fa63d0cabcbf1d7c1f5288c775749e75a23ea3fcc19f24106
+size 1915502

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_816_9528e9623fc2d0e7abfa.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9528e9623fc2d0e7abfaa55399e5bd525687c205d1a0412994f660557f10253f
+size 1949800

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/media/videos/validation_vis/video_0_918_f8392fb64b9a16233691.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f8392fb64b9a1623369104dd5593c97eb46c22f90fd5209b3b0958aea803349e
+size 1928123

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/files/wandb-summary.json

@@ -0,0 +1 @@
+{"_wandb": {"runtime": 98708}}

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/logs/debug.log

@@ -23,3 +23,4 @@ config: {'experiment': {'debug': '${debug}', 'tasks': ['training'], 'num_nodes':
 2026-04-19 21:09:35,867 INFO    MainThread:124537 [wandb_run.py:_redirect():2376] Wrapping output streams.
 2026-04-19 21:09:35,867 INFO    MainThread:124537 [wandb_run.py:_redirect():2401] Redirects installed.
 2026-04-19 21:09:35,878 INFO    MainThread:124537 [wandb_init.py:init():895] run started, returning control to user process
+2026-04-21 00:34:39,733 WARNING MsgRouterThr:124537 [router.py:message_loop():77] message_loop has been closed

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260419_210928-stage_b_joint_offline/run-stage_b_joint_offline.wandb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bdeadb116b3ed8f1bab3d868809af2df0f56b972155d1733b2ea51a488b1892d
-size 21333190
+oid sha256:e340fa6d05e3d789643f4231f17087aab061559208bc81e8bc57ec34931b6f6d
+size 41870251

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/conda-environment.yaml

@@ -0,0 +1,4 @@
+name: worldmem
+channels:
+  - defaults
+prefix: /home/x_fahkh/.conda/envs/worldmem

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_0_d99c30e3f713e6884745.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d99c30e3f713e6884745927c1b2666649f115e9da534ab06ede504b4fb387cc6
+size 2078098

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_102_55531416f1a0f46778e0.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:55531416f1a0f46778e03f40e99036baeee0c8e3e367cdb40c10afa9d6370581
+size 2152198

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_204_b417c4bcea7d8fabb351.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b417c4bcea7d8fabb3510df66fe5669548fa4cc60348bfbb66e29bcf59157d4a
+size 2025535

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_306_90b4cff004dbef7461ac.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:90b4cff004dbef7461ac5583428825dc4a7c6ff5e40196ceabd28292f56c0129
+size 2218786

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_408_5df38886c179815b23f2.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5df38886c179815b23f25c5e00b9f8523c9c4f375fcbe418bd78a9d80d0daa93
+size 1953458

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_510_fcf8664bcf56b448556a.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fcf8664bcf56b448556aebfc07ad13ab458214388d6501141ede0c5a158c94bd
+size 1971189

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_612_ffb9a55464b8c069ef8b.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ffb9a55464b8c069ef8b4adde38b34e565f6dec881100cb6ac89357cf9f46044
+size 2298093

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/media/videos/validation_vis/video_0_714_ac56a01406adfbb1f0de.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac56a01406adfbb1f0de685ea8c24fece03a03032fbad51f052e8b0bf1621939
+size 2013074

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/wandb-metadata.json

@@ -0,0 +1,1228 @@
+{
+    "os": "Linux-5.14.0-611.45.1.el9_7.x86_64-x86_64-with-glibc2.34",
+    "python": "3.10.19",
+    "heartbeatAt": "2026-04-20T22:35:49.700642",
+    "startedAt": "2026-04-20T22:35:48.769720",
+    "docker": null,
+    "cuda": null,
+    "args": [
+        "+name=train_stage_b_hierarchy_bimamba_joint",
+        "algorithm=df_video_mamba3stage",
+        "experiment.num_nodes=1",
+        "dataset.save_dir=/proj/cvl/users/x_fahkh2/WorldMem_Repro/datasets/minecraft",
+        "dataset.n_frames=200",
+        "+dataset.n_frames_valid=200",
+        "+dataset.angle_range=110",
+        "+dataset.pos_range=8",
+        "+dataset.wo_updown=false",
+        "+dataset.customized_validation=true",
+        "+dataset.add_timestamp_embedding=true",
+        "+dataset.use_explicit_memory_frames=false",
+        "algorithm.training_stage=stage_b_diffusion_training",
+        "algorithm.stage_b_joint_training=true",
+        "algorithm.stage_b_memory_aux_weight=0.1",
+        "algorithm.use_mamba_memory_pipeline=true",
+        "algorithm.use_oracle_pose_eval=true",
+        "algorithm.enable_memory_noise_curriculum=false",
+        "+algorithm.use_memory_attention=false",
+        "+algorithm.relative_embedding=false",
+        "algorithm.memory_retrieval_topk=32",
+        "algorithm.diff_window_size=8",
+        "algorithm.memory_condition_length=0",
+        "algorithm.context_frames=100",
+        "+algorithm.n_tokens=8",
+        "experiment.training.lr=2e-5",
+        "experiment.training.batch_size=8",
+        "experiment.training.checkpointing.every_n_train_steps=2500",
+        "experiment.training.max_steps=175000",
+        "experiment.validation.val_every_n_step=2500",
+        "resume=stage_b_joint_offline",
+        "+output_dir=/proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/"
+    ],
+    "state": "running",
+    "program": "-m main",
+    "codePathLocal": null,
+    "git": {
+        "remote": "https://huggingface.co/BonanDing/WorldMem_Repro",
+        "commit": "115f0dc1d074fb06e673cad462ec8179ad88846c"
+    },
+    "email": "fahad.khan@liu.se",
+    "root": "/proj/cvl/users/x_fahkh2/WorldMem_Repro",
+    "host": "node115",
+    "username": "x_fahkh",
+    "executable": "/proj/cvl/users/x_fahkh2/envs/worldmem/bin/python",
+    "cpu_count": 112,
+    "cpu_count_logical": 224,
+    "cpu_freq": {
+        "current": 1071.4902723214284,
+        "min": 800.0,
+        "max": 3800.0
+    },
+    "cpu_freq_per_core": [
+        {
+            "current": 3670.893,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3799.961,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3452.305,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3799.961,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3333.398,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3794.647,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3757.541,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3799.961,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.038,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3799.961,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3804.222,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3804.507,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.9,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3802.739,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 3799.961,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        },
+        {
+            "current": 800.0,
+            "min": 800.0,
+            "max": 3800.0
+        }
+    ],
+    "disk": {
+        "/": {
+            "total": 30.58770751953125,
+            "used": 16.404800415039062
+        }
+    },
+    "gpu": "NVIDIA H200",
+    "gpu_count": 8,
+    "gpu_devices": [
+        {
+            "name": "NVIDIA H200",
+            "memory_total": 150754820096
+        },
+        {
+            "name": "NVIDIA H200",
+            "memory_total": 150754820096
+        },
+        {
+            "name": "NVIDIA H200",
+            "memory_total": 150754820096
+        },
+        {
+            "name": "NVIDIA H200",
+            "memory_total": 150754820096
+        },
+        {
+            "name": "NVIDIA H200",
+            "memory_total": 150754820096
+        },
+        {
+            "name": "NVIDIA H200",
+            "memory_total": 150754820096
+        },
+        {
+            "name": "NVIDIA H200",
+            "memory_total": 150754820096
+        },
+        {
+            "name": "NVIDIA H200",
+            "memory_total": 150754820096
+        }
+    ],
+    "memory": {
+        "total": 2015.559253692627
+    }
+}

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/files/wandb-summary.json

@@ -0,0 +1 @@
+{"_wandb": {"runtime": 62399}}

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/logs/debug.log

@@ -0,0 +1,26 @@
+2026-04-21 00:35:48,797 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Current SDK version is 0.17.9
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Configure stats pid to 1288079
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Loading settings from /home/x_fahkh/.config/wandb/settings
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Loading settings from /proj/cvl/users/x_fahkh2/WorldMem_Repro/wandb/settings
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Loading settings from environment variables: {'disabled': 'true'}
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Applying setup settings: {'_disable_service': False}
+2026-04-21 00:35:48,798 WARNING MainThread:1288079 [wandb_setup.py:_flush():77] Could not find program at -m main
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_setup.py:_flush():77] Inferring run settings from compute environment: {'program_relpath': None, 'program': '-m main'}
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:_log_setup():524] Logging user logs to /proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/logs/debug.log
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:_log_setup():525] Logging internal logs to /proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/logs/debug-internal.log
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:init():608] calling init triggers
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:init():615] wandb.init called with sweep_config: {}
+config: {'experiment': {'debug': '${debug}', 'tasks': ['training'], 'num_nodes': 1, 'training': {'precision': '16-mixed', 'compile': False, 'lr': 2e-05, 'batch_size': 8, 'max_epochs': -1, 'max_steps': 175000, 'max_time': None, 'data': {'num_workers': 4, 'shuffle': True}, 'optim': {'accumulate_grad_batches': 1, 'gradient_clip_val': 1.0}, 'checkpointing': {'every_n_train_steps': 2500, 'every_n_epochs': None, 'train_time_interval': None, 'enable_version_counter': False}}, 'validation': {'precision': '16-mixed', 'compile': False, 'batch_size': 4, 'val_every_n_step': 2500, 'val_every_n_epoch': None, 'limit_batch': 1, 'inference_mode': False, 'data': {'num_workers': 4, 'shuffle': False}}, 'test': {'precision': '16-mixed', 'compile': False, 'batch_size': 1, 'limit_batch': 1, 'inference_mode': False, 'data': {'num_workers': 4, 'shuffle': False}}, 'logging': {'metrics': None}, '_name': 'exp_video'}, 'dataset': {'debug': '${debug}', 'metadata': 'data/${dataset.name}/metadata.json', 'data_mean': 0.5, 'data_std': 0.5, 'save_dir': '/proj/cvl/users/x_fahkh2/WorldMem_Repro/datasets/minecraft', 'n_frames': 200, 'context_length': 1, 'resolution': 128, 'observation_shape': [3, '${dataset.resolution}', '${dataset.resolution}'], 'external_cond_dim': 0, 'validation_multiplier': 1, 'frame_skip': 1, 'action_cond_dim': 25, '_name': 'video_minecraft', 'n_frames_valid': 200, 'angle_range': 110, 'pos_range': 8, 'wo_updown': False, 'customized_validation': True, 'add_timestamp_embedding': True, 'use_explicit_memory_frames': False}, 'algorithm': {'debug': '${debug}', 'lr': '${experiment.training.lr}', 'x_shape': '${dataset.observation_shape}', 'frame_stack': 1, 'frame_skip': '${dataset.frame_skip}', 'data_mean': '${dataset.data_mean}', 'data_std': '${dataset.data_std}', 'external_cond_dim': 0, 'context_frames': 100, 'weight_decay': 0.002, 'warmup_steps': 1000, 'optimizer_beta': [0.9, 0.99], 'uncertainty_scale': 1, 'guidance_scale': 0.0, 'chunk_size': 1, 'scheduling_matrix': 'autoregressive', 'noise_level': 'random_all', 'causal': True, 'diffusion': {'objective': 'pred_v', 'beta_schedule': 'sigmoid', 'schedule_fn_kwargs': {}, 'clip_noise': 20.0, 'use_snr': False, 'use_cum_snr': False, 'use_fused_snr': True, 'snr_clip': 5.0, 'cum_snr_decay': 0.96, 'timesteps': 1000, 'sampling_timesteps': 20, 'ddim_sampling_eta': 0.0, 'stabilization_level': 15, 'architecture': {'network_size': 64, 'attn_heads': 4, 'attn_dim_head': 64, 'dim_mults': [1, 2, 4, 8], 'resolution': '${dataset.resolution}', 'attn_resolutions': [16, 32, 64, 128], 'use_init_temporal_attn': True, 'use_linear_attn': True, 'time_emb_type': 'rotary'}}, 'n_frames': '${dataset.n_frames}', 'metadata': '${dataset.metadata}', 'action_cond_dim': 25, 'use_plucker': True, 'memory_condition_length': 0, 'log_video': True, 'use_mamba_memory_pipeline': True, 'training_stage': 'stage_b_diffusion_training', 'stage_b_joint_training': True, 'stage_b_memory_aux_weight': 0.1, 'stage_b_prefix_corruption_prob': 0.0, 'stage_b_prefix_corruption_timestep_range': [5, 50], 'stage_b_prefix_corruption_max_recent_frames': 32, 'diff_window_size': 8, 'use_precomputed_features': False, 'mamba_latent_channels': 16, 'mamba_model_dim': 256, 'mamba_depth': 4, 'mamba_cond_dim': 256, 'mamba_d_state': 16, 'mamba_d_conv': 4, 'mamba_expand': 2, 'memory_scene_grid': [2, 4], 'memory_num_tokens': 4, 'memory_retrieval_topk': 32, 'allow_mamba_fallback': False, 'strict_causal_training': True, 'strict_causal_evaluation': True, 'use_oracle_pose_eval': True, 'enable_memory_noise_curriculum': False, 'curriculum_phase_boundaries': [0.2, 0.7], 'curriculum_noise_ranges': [[600, 1000], [200, 900], [0, 400]], 'curriculum_horizons': [50, 100, 200], '_name': 'df_video_mamba3stage', 'use_memory_attention': False, 'relative_embedding': False, 'n_tokens': 8}, 'debug': False, 'wandb': {'entity': 'turlin', 'project': 'worldmem', 'mode': 'online'}, 'resume': 'stage_b_joint_offline', 'load': None, 'name': 'train_stage_b_hierarchy_bimamba_joint', 'output_dir': '/proj/cvl/users/x_fahkh2/WorldMem_Repro/checkpoints/hierarchy_bimamba_stage_b_joint_fixed/'}
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:init():658] starting backend
+2026-04-21 00:35:48,798 INFO    MainThread:1288079 [wandb_init.py:init():662] setting up manager
+2026-04-21 00:35:48,813 INFO    MainThread:1288079 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
+2026-04-21 00:35:48,820 INFO    MainThread:1288079 [wandb_init.py:init():670] backend started and connected
+2026-04-21 00:35:49,492 INFO    MainThread:1288079 [wandb_init.py:init():768] updated telemetry
+2026-04-21 00:35:49,553 INFO    MainThread:1288079 [wandb_init.py:init():801] communicating run to backend with 90.0 second timeout
+2026-04-21 00:35:49,557 INFO    MainThread:1288079 [wandb_init.py:init():852] starting run threads in backend
+2026-04-21 00:35:58,257 INFO    MainThread:1288079 [wandb_run.py:_console_start():2465] atexit reg
+2026-04-21 00:35:58,257 INFO    MainThread:1288079 [wandb_run.py:_redirect():2311] redirect: wrap_raw
+2026-04-21 00:35:58,257 INFO    MainThread:1288079 [wandb_run.py:_redirect():2376] Wrapping output streams.
+2026-04-21 00:35:58,257 INFO    MainThread:1288079 [wandb_run.py:_redirect():2401] Redirects installed.
+2026-04-21 00:35:58,279 INFO    MainThread:1288079 [wandb_init.py:init():895] run started, returning control to user process
+2026-04-21 17:55:51,088 WARNING MsgRouterThr:1288079 [router.py:message_loop():77] message_loop has been closed

hierarchy_bimamba_stage_b_joint_fixed/wandb/offline-run-20260421_003548-stage_b_joint_offline/run-stage_b_joint_offline.wandb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ad57f7f5f8e9fc05c90ff80adbeb4eb0de562b42b207fef9c3110535c87f3133
+size 26547956