VideoMAE model does not accept batch size > 1


I was trying to make the VideoMAE model work with a batch size > 1 but I see this is not possible. Is this the expected behaviour?

I believe there is a similar issue with no answer.

Thank you.


Thanks for your interest in VideoMAE! I’ve replied on the linked issue.