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

Type d'évaluation

Mise en situation réelle

Analyse

Intelligence Artificielle

Compétences ciblées

gpu
device-management
bugfix
training
pytorch

Scénario Technique

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.

Prêt à évaluer ce talent ?

Accédez à l'interface de test complète pour soumettre votre architecture ou votre code.

Commencer le Défis
    PyTorch - Correction d’un modèle avec mauvaise gestion du device (CPU/GPU) | Test Technique & Recrutement