hustvl/QueryInst

gpu

Opened this issue · 2 comments

roar-1128 commented

请问一张v100能训练吗

Yuxin-CV commented

Hi，一张卡跑COCO恐怕不行。

但如果您跑的数据集规模比较小，可以尝试用我们提供的COCO pretrain model去做单卡的finetune。

116022017144 commented

我正在用一张A100 40G来训练r50_fpn_1x_coco,似乎需要三天