search

Kay1794 / MMVP-motion-matrix-based-video-prediction

This is the official repo of MMVP: motion-matrix-based video prediction (ICCV 2023)
MIT License
36 stars 1 forks source link

How many gpu memory do I need to train the model from scratch? #4

Open Kendrick-Stein opened 1 year ago

Kay1794 commented 1 year ago

If it is for the UCF Sports dataset, we can set the batch size to 2 on a single 1080Ti. But you may also decrease the length of image features or prediction features length to reduce the model consumption.