search

Dawn-LX / OpenVoc-VidVRD

Official code for the ICLR2023 paper Compositional Prompt Tuning with Motion Cues for Open-vocabulary Video Relation Detection
41 stars 1 forks source link

GPU #6

Open ballin0105 opened 11 months ago

ballin0105 commented 11 months ago

Hi,

I'm curious about the specific GPU model you utilized for training during your research.

Thank you!

Dawn-LX commented 11 months ago

refer to here:

https://github.com/Dawn-LX/OpenVoc-VidVRD/blob/3cb19017f76c614f9c9a494efb19f075cf51a69e/tools/train_relation_cls.py#L715

I wrote the GPU requirements for each exp in the training script.