Correction d’un modèle avec mauvaise gestion du device (CPU/GPU)

Scénario

Contexte : Le modèle s’exécute partiellement sur GPU et partiellement sur CPU. Le Problème : Des erreurs apparaissent liées à des tenseurs sur devices différents. Extrait logique actuelle : model.to(device) Sans déplacer les données. Contraintes : Assurer cohérence device pour modèle et batchs. Vérifier gestion des tenseurs. Livrable attendu : Corriger le pipeline pour garantir exécution cohérente sur CPU ou GPU.

Compétences ciblées