TraceML, un outil de diagnostic d'entraînement PyTorch open-source, a été utilisé pour reproduire une régression connue dans l'entraînement de LeRobot et mesurer l'impact de la correction amont #2408. L'analyse a révélé que bien que la plupart des récupérations du DataLoader soient rapides, les récupérations récurrentes prenaient auparavant plus de cinq secondes.

  • Le temps par étape a diminué d'environ 1500ms à 117ms après la correction.
  • Le temps de récupération du CPU DataLoader est passé de ~1375ms à 1,5ms pour le premier couple et de ~1427ms à 1,3ms pour le second couple.
  • Le temps de calcul est resté stable autour de 110–116ms, confirmant que la charge de travail a changé de INPUT-BOUND à COMPUTE-BOUND.

L'outil aide à identifier où le temps supplémentaire est perdu lorsque l'entraînement ralentit après un changement, en instrumentant la boucle d'entraînement et en comparant les timings entre deux exécutions.