Update config.yaml

config.yaml

@@ -8,7 +8,7 @@ wandb_entity: jinhuiye
 wandb_project: InternVLA
 is_debug: false
 framework:
-  framework_py: DinoQFormerACT
+  framework_py: InternVLA-M1
   qwenvl:
     base_vlm: /mnt/phwfile/efm_t/zhuyangkun_tmp_need_del/exp/exp_08_09/manip_sys2_qwen25_3b_onevision_molmo_a0all_refsp20/checkpoint-20000
     attn_implementation: flash_attention_2
@@ -19,6 +19,8 @@ framework:
     qformer_end_layer: 37
     qformer_start_layer: 36
     num_query_tokens: 64
+    input_dim: 2048
+    ouptput_dim: 768
     grad_scale: 0.5
   action_model:
     action_model_type: DiT-B
@@ -89,4 +91,4 @@ trainer:
   enable_gradient_checkpointing: true
   enable_mixed_precision_training: true
 is_resume: false
-output_dir: ./results/Checkpoints/0906_bestvla_retrain_lr_v2
+output_dir: ./results/Checkpoints/0906_internvla_m1