Problème alimentation d'un Transformer avec une matrice

a marqué ce sujet comme résolu.

Bonjour, J’ai un dossier qui comporte des images de 32 x 32 pixels en niveaux de gris de caractères manuscrits. Je divise chaque image en segments de taille 8x8 pixels. Puis chaque segment est ensuite divisé en sous-segments de taille 2x2 pixels. Je conserve l’emplacement de l’image d’origine pour chaque sous-segment. Normalement je vais avoir comme résultat une matrice de 16 lignes et 16 colonnes dont chaque élément est un sous-segment de 2 x 2 pixels. C’est pour cela j’ai crée un Transformer qui accepte en entrée une matrice de 16 x 16. A chaque fois que je commence à entrainer ce modèle, une erreur survient: "RuntimeError: mat1 and mat2 shapes cannot be multiplied (1x1024 and 2x256)". Donner moi un coup de main svp.

Connectez-vous pour pouvoir poster un message.
Connexion

Pas encore membre ?

Créez un compte en une minute pour profiter pleinement de toutes les fonctionnalités de Zeste de Savoir. Ici, tout est gratuit et sans publicité.
Créer un compte