semantic-segmentation-pytorch
semantic-segmentation-pytorch copied to clipboard
Inference on multiple gpus
Is it possible to perform inference/test on multiple gpus. Right now the test script seems to accepts only one gpu id and testing on large dataset is very slow.
Yes, it should be very straight forward if you write a script similar to eval_multipro.py
, which is for multi-gpu evaluation.