TraceML, alat diagnosis pelatihan PyTorch sumber terbuka, digunakan untuk mereproduksi regresi pelatihan LeRobot yang diketahui dan mengukur dampak perbaikan hulu #2408. Analisis mengungkapkan bahwa meskipun sebagian besar pengambilan DataLoader cepat, pengambilan berulang sebelumnya memakan waktu lebih dari lima detik.
- Waktu langkah menurun dari sekitar 1500ms menjadi 117ms setelah perbaikan.
- Waktu pengambilan CPU DataLoader turun dari ~1375ms menjadi 1.5ms pada pasangan pertama dan dari ~1427ms menjadi 1.3ms pada pasangan kedua.
- Waktu komputasi tetap stabil di sekitar 110–116ms, mengonfirmasi bahwa beban kerja bergeser dari INPUT-BOUND ke COMPUTE-BOUND.
Alat ini membantu mengidentifikasi di mana waktu tambahan hilang ketika pelatihan melambat setelah perubahan dengan menginstrumentasi loop pelatihan dan membandingkan waktu antara dua eksekusi.