Partage
  • Partager sur Facebook
  • Partager sur Twitter

Alimenter un Transformer avec une matrice

    16 mars 2024 à 11:37:37

    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.
    • Partager sur Facebook
    • Partager sur Twitter

    Alimenter un Transformer avec une matrice

    × Après avoir cliqué sur "Répondre" vous serez invité à vous connecter pour que votre message soit publié.
    • Editeur
    • Markdown