Multi-GPU Testing Support?

speedinghzl / pytorch-segmentation-toolbox

PyTorch Implementations for DeeplabV3 and PSPNet

MIT License

772 stars 167 forks source link

Multi-GPU Testing Support? #22

Closed JiYuanFeng closed 5 years ago

JiYuanFeng commented 5 years ago

Can this toolbox support the Multi-scale testing?

JiYuanFeng commented 5 years ago

Sorry ! I think I am wrong. actually i am wondering to know is the project support Multi-GPU testing?

speedinghzl commented 5 years ago

No, does not support. You are welcome to add a new feature to support Multi-GPU testing.

aasharma90 commented 5 years ago

Hi @JiYuanFeng @speedinghzl ,

I've attempted to create one. But, please use it with caution, since I'm a newbie to PyTorch and could've implemented it sub-optimally. https://www.dropbox.com/s/ryf1f4507z4a5w4/evaluate_multigpu.py?dl=0 I'll appreciate if you could find any potential bugs and update me. Thanks!

ps: I've checked that it gives the same performance results as evaluate.py (i.e. with single gpu testing), and all the 4 GPUs were occupied (with BATCH_SIZE=4, and GPU_IDs='0,1,2,3')

speedinghzl commented 5 years ago

@aasharma90 Nice work!