pipeline is really slow - consulting [D]

En 3 lignesUtilisateur demande conseil sur un goulot d'étranglement en entraînement d'imitation learning robotique. Pipeline : 4 caméras RGB 128×128 → ResNet18 gelé → DiT (~50M params, 8 couches) prédisant chunks d'actions. GPU A4500 à 20-30% d'utilisation, CPU saturé, ~10 iter/sec. Profiler montre optimizer_step dominant (62.4%).

Résumé généré par Claude — vérifié par l'humain