search

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 567 forks source link

Why does mynn.DataParallel not support multiple gpus? #209

Closed ashnair1 closed 4 years ago

ashnair1 commented 5 years ago

In infer_simple.py the mask RCNN model is created as follows: 

maskRCNN = mynn.DataParallel(maskRCNN, cpu_keywords=['im_info', 'roidb'],
                                 minibatch=True, device_ids=[0])  # only support single GPU

I've noticed that the DataParallel module only supports a single GPU. I actually wanted to speed up my inference time i.e. time taken to visualise the detections of the model on an image since I have upward of 1000 images. Is there a way to visualise and save the detections on multiple images using multiple gpus? Also why can't the model support multiple GPUs?

1179021477 commented 4 years ago

Excuse me, have you solved this problem?

ashnair1 commented 4 years ago

No, I actually just moved onto detectron2