Erreur pour utiliser resnet18 avec MNIST

Question

J'ai essayé d'adapter le modèle resnet18 comme indiqué dans les diapos mais j'ai l'erreur suivante :

RuntimeError: Given groups=1, weight of size [64, 1, 7, 7], expected input[1, 10000, 28, 28] to have 1 channels, but got 10000 channels instead

Est-il nécessaire de modifier les données ??
Pour l'instant, j'ai modifié la couche d'entrée et celle de sortie, comme sur les diapos.

Answer 1

Vous devez faire en sorte que la taille de vos données d'entrée correspond à ce qu'attend votre modèle (ou l'inverse). Libre à vous de modifier l'un ou l'autre (ou les 2).