Fixed bug

Signed-off-by: Seongwoo Kim <[email protected]>

Author: Seongoo

physical_ai_server/physical_ai_server/training/trainers/lerobot/lerobot_trainer.py

@@ -171,7 +171,7 @@ def train(self, cfg: TrainPipelineConfig, stop_event=None):
             if stop_event and stop_event.is_set():
                 self.logger.info('Training stopped by stop event')
                 break
-            self.current_step = step
+            self.current_step = step + 1
             start_time = time.perf_counter()
             batch = next(dl_iter)
             train_tracker.dataloading_s = time.perf_counter() - start_time