how much video memory is needed for project reasoning

[bboyalong01]

Hello, how much video memory is needed for the project to run?

[moonbow721]

Hi, thanks for your interest in our work!

Regarding your question about VRAM:

Inference: Based on my testing, the inference scripts t2m_infer.sh and tm2m_infer.sh require less than 16GB VRAM on a single GPU. You can potentially reduce memory usage further by decreasing the batch_size in the configuration file.

Training: For reference, we utilized 8xH100 GPUs for the training process.

Hope this helps!