why using only first element of batch?

[fcakyon]

Thanks for the work!

If I understand correctly, you only use the first element of the input batch on MViT class here:

MeMViT/memvit/models/video_model_builder.py

Line 1068 in 20a24f0

x = x[0]

What is the reason for that? What am I missing?