TraceML, an open-source PyTorch training diagnosis tool, was used to reproduce a known LeRobot training regression and measure the impact of upstream fix #2408. The analysis revealed that while most DataLoader fetches were quick, recurring fetches previously took over five seconds.
- Step time decreased from approximately 1500ms to 117ms after the fix.
- CPU DataLoader fetch time dropped from ~1375ms to 1.5ms in the first pair and from ~1427ms to 1.3ms in the second pair.
- Compute time remained stable around 110–116ms, confirming the workload shifted from INPUT-BOUND to COMPUTE-BOUND.
The tool helps identify where extra time goes when training slows down after a change by instrumenting the training loop and comparing timings between two runs.