error: not enough values to unpack (expected 3, got 2)

[liuqiwei1028]

发生报错：
2026-02-20 08:15:48 - tokenizer_training_rank_0 - INFO - [Epoch 1/12, Step 594700/594705] LR: 0.000000, Loss: -0.0286
[Epoch 1/1, Step 594700/594705] LR: 0.000000, Loss: -0.0286
2026-02-20 08:15:48 - tokenizer_training_rank_0 - INFO - - VQ Loss: -0.0705

• Recon Loss Pre: 0.0066
• Recon Loss All: 0.0054
• VQ Loss: -0.0705
• Recon Loss Pre: 0.0066
• Recon Loss All: 0.0054
  Error occurred during training: not enough values to unpack (expected 3, got 2)
  File "D:\Projects\stocktrain\Kronos\finetune_csv..\model\module.py", line 331, in forward
  batch_size, seq_len, _ = x.shape
  ^^^^^^^^^^^^^^^^^^^^^^
  ValueError: not enough values to unpack (expected 3, got 2)
  Training failed!

finetune_tokenizer.py line 193 242 进行修改ori_batch_x = ori_batch_x.to(device, non_blocking=True)
请问下这里为啥要squeeze(0)？