Penulis telah mengintegrasikan alat diagnostik ke dalam TraceML, sebuah proyek observabilitas PyTorch sumber terbuka, untuk membantu pengguna membedakan antara masalah kinerja pelatihan yang dibatasi oleh komputasi GPU dan yang dibatasi oleh input. Fitur baru ini memungkinkan pengembang dengan cepat mengidentifikasi apakah utilisasi GPU yang rendah disebabkan oleh pemuatan data yang lambat atau batas komputasi model yang sebenarnya.
- Integrasi menambahkan `TraceMLTrainerCallback` ke `Trainer` Hugging Face, memerlukan perubahan pengaturan minimal.
- Ini memisahkan waktu tunggu input dari pekerjaan langkah untuk memberikan putusan praktis tentang jenis bottleneck.
- Contoh Colab dengan ResNet-50 menunjukkan bahwa penyesuaian pengaturan DataLoader mengubah diagnosis dari dibatasi input menjadi dibatasi komputasi dan meningkatkan kecepatan sebesar 1.83×.
Alat ini membantu pengguna menghindari mengoptimalkan komponen yang salah dengan memperjelas batasan spesifik yang mempengaruhi jalankan pelatihan mereka.