Problem about using multiple GPUs for one-stage methods

Jingkang50 / OpenPSG

Benchmarking Panoptic Scene Graph Generation (PSG), ECCV'22

https://psgdataset.org

MIT License

416 stars 68 forks source link

Problem about using multiple GPUs for one-stage methods #86

Closed Fillip1233 closed 1 year ago

Fillip1233 commented 1 year ago

Hi,author!I encountered a problem when using multi gpu training.If I want to specify the gpu card number for training, how should I set it?I can't seem to find a solution.In addition, the multi gpu training instructions you provide on readme use distributed training. Is the -- gpus 8 parameter configuration ineffective? thank you!

Hanzy1996 commented 1 year ago

I have the same question. Could you please share how you solve it?

Best Regards.