debug(inference): 添加推理阶段qpos归一化

2026-02-06 09:00:44 +08:00
parent b0a944f7aa
commit 66009473ad
7 changed files with 859 additions and 121 deletions
@@ -4,10 +4,10 @@ defaults:
  - data: resnet_dataset

 train:
-  batch_size: 8       # Batch size for training
+  batch_size: 32       # Batch size for training
  lr: 1e-4            # Learning rate
-  max_steps: 10000    # Maximum training steps
+  max_steps: 20000    # Maximum training steps
  log_freq: 100       # Log frequency (steps)
-  save_freq: 1000     # Save checkpoint frequency (steps)
+  save_freq: 2000     # Save checkpoint frequency (steps)
  device: "cuda"      # Device: "cuda" or "cpu"
  num_workers: 8      # DataLoader workers (set to 0 for debugging, 8 for production)