fix save check point

This commit is contained in:
vincentccc
2025-09-08 23:21:03 +08:00
parent 64b41cc4f6
commit 49af1cf154
2 changed files with 6 additions and 3 deletions
+3 -3
View File
@@ -5,9 +5,8 @@
log_name: "robotic_training"
log_project: "vla_training"
model_type: qwen2_5
pretrained_qwen_vl_path: "/path/to/qwen_vl_model/"
# whether to enable fast tokenizer
use_fast_tokenizer: false
pretrained_qwen_vl_path: "/path/to/wallx_model/"
use_fast_tokenizer: false # True: train FAST, False: train Flow
action_tokenizer_path: "/path/to/fast/"
save_path: "/path/to/workspace/"
@@ -27,6 +26,7 @@ num_epoch: 100
gradient_accumulation_steps: 32
batch_size_per_gpu: 8
padding_side: left
epoch_save_interval: 10
# Robot configuration - Define degrees of freedom for each component
dof_config: