jy0205 / STCAT

[NeurIPS 2022] Embracing Consistency: A One-Stage Approach for Spatio-Temporal Video Grounding

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Question About Training

Simoyuan opened this issue · comments

Hi, when I train the code in hc-stvg, where the resolution is 448. and is trained in 32 gpus. But I find my cuda allways is out of memory. My config yaml follows yours. Can you answer the question? Thank you.

You can reduce the input resolution, e.g, res=416 or 384