search

alibaba-mmai-research / HyRSM

Code for our CVPR 2022 Paper "Hybrid Relation Guided Set Matching for Few-shot Action Recognition".
Apache License 2.0
26 stars 5 forks source link

How to train on GPUs with restricted VRAM #3

Open luowyang opened 1 year ago

luowyang commented 1 year ago

It seems that the code cannot run on mid-end GPUs, such as NVIDIA RTX 2060s, without OOM. How can we modify the code to reduce the VRAM consumption per GPU?

wangxiang1230 commented 1 year ago

It seems that the code cannot run on mid-end GPUs, such as NVIDIA RTX 2060s, without OOM. How can we modify the code to reduce the VRAM consumption per GPU?

It is recommended that you can reduce the query number (i.e., "QUERY_PER_CLASS" in yaml file) to accommodate fewer computing resources. But the performance may be different from the default setting.