MCG-NJU/VideoMAE

The GPU memory usage of UCF and Kinetics is different.

Backdrop9019 opened this issue · 0 comments

Hello, thank you for the kind words about the code.
When fine-tuning using your code on both UCF and Kinetics datasets, I've observed a significant difference in memory usage. Despite using the same number of frames (16 frames), batch size, and dataset class for both, the GPU memory consumption is almost three times higher for Kinetics than UCF. Specifically, with a batch size of 16 on an RTX 3090, Kinetics consumes about 22GB while UCF only uses around 8GB. Can you explain the reason for this discrepancy?