joeyz0z / MeaCap

(CVPR2024) MeaCap: Memory-Augmented Zero-shot Image Captioning
37 stars 1 forks source link

IndexError: The shape of the mask [1, 6] at index 1 does not match the shape of the indexed tensor [1, 7, 50265] at index 1 #8

Open 1013990424 opened 2 months ago

1013990424 commented 2 months ago

Great job, thank you for sharing, but I encountered this problem while predicting my own pictures. Do you have any good suggestions for modification?

IndexError: The shape of the mask [1, 6] at index 1 does not match the shape of the indexed tensor [1, 7, 50265] at index 1