issue on shape of underlying tensors

[Olivier-aka-Raiden]

Hi, I wanted to give it a try and when trying to do inferance on the base model, I get an error as follows :
IndexError: The shape of the mask [1, 4] at index 0 does not match the shape of the indexed tensor [8, 5] at index 0

I tried with jpg and png image, both have the exact output. Also, using the python example code didn't change anything, error is the same.
Is it because of new tensorflow version ? Can you help me fix the code to make it work ? Probably the definition of the dtype is incorrect.

[rsommerfeld]

Hi, have you been able to resolve your problem? If not, can you provide more details? All parts of the repository are still working on a clean setup for me. The repository also relies on pytorch, not tensorflow.