Upload checkpoints_vlm_gym_jigsaw_one_image_lr2e_5_mse_only_ema9999_hashed/checkpoints_vlm_gym_jigsaw_one_image_lr2e_5_mse_only_ema9999_hashed

checkpoints_vlm_gym_jigsaw_one_image_lr2e_5_mse_only_ema9999_hashed/checkpoints_vlm_gym_jigsaw_one_image_lr2e_5_mse_only_ema9999_hashed/wandb/offline-run-20260111_235518-vlm_gym_jigsaw_one_img_lr2e_5_mse_only_ema9999_hashed-run0/files/config.yaml

@@ -0,0 +1,181 @@
+wandb_version: 1
+
+_wandb:
+  desc: null
+  value:
+    python_version: 3.11.10
+    cli_version: 0.23.1
+    framework: huggingface
+    huggingface_version: 4.49.0
+    is_jupyter_run: false
+    is_kaggle_kernel: false
+    start_time: 1768175718
+    t:
+      1:
+      - 1
+      - 5
+      - 11
+      - 41
+      - 49
+      - 53
+      - 71
+      - 105
+      2:
+      - 1
+      - 5
+      - 11
+      - 41
+      - 49
+      - 53
+      - 71
+      - 105
+      3:
+      - 4
+      - 13
+      - 14
+      - 37
+      - 42
+      4: 3.11.10
+      5: 0.23.1
+      6: 4.49.0
+      13: linux-x86_64
+    e:
+      p78nkupquzvg4sp7iakb8i8e0ffje8id:
+        os: Linux-6.6.93+-x86_64-with-glibc2.35
+        python: CPython 3.11.10
+        started_at: '2026-01-11T23:55:18.222054Z'
+        args:
+        - --dataset_config_file
+        - ./data/configs/vlm_gym_jigsaw_train_mseloss_only.yaml
+        - --eval_dataset_config_file
+        - ./data/configs/vlm_gym_jigsaw_train_mseloss_only.yaml
+        - --viz_dataset_config_file
+        - ./data/configs/vlm_gym_jigsaw_train_mseloss_only.yaml
+        - --inference_hash_file
+        - /home/clouduser/Code/Github/launch_new/hashes_test_set_v10.json
+        - --train_data_dir
+        - /home/clouduser/Code/data/gym/jigsaw-swap_v5/train/
+        - --train_jsonl_path
+        - /home/clouduser/Code/data/gym/jigsaw-swap_v5/train/
+        - --eval_data_dir
+        - /home/clouduser/Code/data/gym/jigsaw-swap_v5/val/
+        - --eval_jsonl_path
+        - /home/clouduser/Code/data/gym/jigsaw-swap_v5/val/
+        - --model_path
+        - /home/clouduser/Code/Models/BAGEL-7B-MoT
+        - --layer_module
+        - Qwen2MoTDecoderLayer
+        - --max_latent_size
+        - '64'
+        - --resume-from
+        - /home/clouduser/Code/Models/BAGEL-7B-MoT
+        - --finetune_from_hf
+        - 'True'
+        - --auto_resume
+        - 'False'
+        - --resume-model-only
+        - 'True'
+        - --finetune-from-ema
+        - 'True'
+        - --log_every
+        - '1'
+        - --lr
+        - 2e-5
+        - --warmup_steps
+        - '300'
+        - --lr_scheduler
+        - cosine
+        - --num_worker
+        - '1'
+        - --expected_num_tokens
+        - '20000'
+        - --max_num_tokens
+        - '20000'
+        - --max_num_tokens_per_sample
+        - '20000'
+        - --visual_und
+        - 'True'
+        - --save_every
+        - '2500'
+        - --total_steps
+        - '5000'
+        - --text_cond_dropout_prob
+        - '0.0'
+        - --vae_cond_dropout_prob
+        - '0.0'
+        - --vit_cond_dropout_prob
+        - '0.0'
+        - --ema
+        - '0.9999'
+        - --checkpoint_dir
+        - /dev/shm/models/checkpoints_vlm_gym_jigsaw_one_image_lr2e_5_mse_only_ema9999_hashed
+        - --wandb_project
+        - bagel
+        - --wandb_name
+        - vlm_gym_jigsaw_one_img_lr2e_5_mse_only_ema9999_hashed
+        - --wandb_dir
+        - /dev/shm/models/checkpoints_vlm_gym_jigsaw_one_image_lr2e_5_mse_only_ema9999_hashed
+        - --wandb_offline
+        - 'True'
+        program: /home/clouduser/Code/Github/unified_world_model/train/pretrain_unified_navit.py
+        code_path: train/pretrain_unified_navit.py
+        code_path_local: train/pretrain_unified_navit.py
+        git:
+          remote_url: https://github.com/para-lost/unified_world_model
+          commit: b61f53a430c22fa4a17c69d2f903ce1fe266f48e
+        root: /dev/shm/models/checkpoints_vlm_gym_jigsaw_one_image_lr2e_5_mse_only_ema9999_hashed
+        host: junyizhang-launch-new-221746038-1-0
+        executable: /opt/conda/bin/python3.11
+        cpu_count: 48
+        cpu_count_logical: 96
+        gpu_type: NVIDIA A100-SXM4-80GB
+        gpu_count: 8
+        disk:
+          /:
+            total: '1052461830144'
+            used: '261625344000'
+        memory:
+          total: '1437332606976'
+        gpu_nvidia:
+        - name: NVIDIA A100-SXM4-80GB
+          memory_total: '85899345920'
+          cuda_cores: 6912
+          architecture: Ampere
+          uuid: GPU-5c8a9afd-07c9-5b67-4b7d-2d070676ab83
+        - name: NVIDIA A100-SXM4-80GB
+          memory_total: '85899345920'
+          cuda_cores: 6912
+          architecture: Ampere
+          uuid: GPU-f44fe102-70f0-4db8-5fd1-c30bfd0f9bb8
+        - name: NVIDIA A100-SXM4-80GB
+          memory_total: '85899345920'
+          cuda_cores: 6912
+          architecture: Ampere
+          uuid: GPU-97f9aee6-0f04-ce00-5732-7d6cab1f9170
+        - name: NVIDIA A100-SXM4-80GB
+          memory_total: '85899345920'
+          cuda_cores: 6912
+          architecture: Ampere
+          uuid: GPU-e645965f-7741-e8d5-63df-bf1e988d5549
+        - name: NVIDIA A100-SXM4-80GB
+          memory_total: '85899345920'
+          cuda_cores: 6912
+          architecture: Ampere
+          uuid: GPU-4123c973-b472-4c08-392e-9df433596e67
+        - name: NVIDIA A100-SXM4-80GB
+          memory_total: '85899345920'
+          cuda_cores: 6912
+          architecture: Ampere
+          uuid: GPU-8a672f68-5ec7-eb5e-8f00-748126c6ce99
+        - name: NVIDIA A100-SXM4-80GB
+          memory_total: '85899345920'
+          cuda_cores: 6912
+          architecture: Ampere
+          uuid: GPU-81b788fb-644b-899f-258e-b2ac8ffffb2b
+        - name: NVIDIA A100-SXM4-80GB
+          memory_total: '85899345920'
+          cuda_cores: 6912
+          architecture: Ampere
+          uuid: GPU-15e8faa7-0d4d-86aa-05db-cf5fb77e5111
+        cuda_version: '12.2'
+        writer_id: p78nkupquzvg4sp7iakb8i8e0ffje8id