0 votes
par dans 10 - Graph Neural Network (GNN)
Bonjour,

Pourquoi faut-il envoyer le AtomEncoder sur le device dans evaluate et train ? N'est-il pas déjà partie prenante du "model" envoyé sur device ?

1 Réponse

0 votes
par Vétéran du GPU 🐋 (8.6k points)
sélectionné par
 
Meilleure réponse
Bonjour,

Il s'agit d'une coquille.
L'atomencoder n'a pas besoin d'être défini dans les fonctions train et test.

Merci de nous l'avoir signalé.
...