search

imatge-upc / rvos

RVOS: End-to-End Recurrent Network for Video Object Segmentation (CVPR 2019)
https://imatge-upc.github.io/rvos/
Other
277 stars 54 forks source link

Questions about multi-gpu #4

Closed jiady1990 closed 5 years ago

jiady1990 commented 5 years ago

Thank you for your amazing work and project! Have you tried to run the project on multi-gpu?(It seems that the released project doesn't support multi-gpu)

carlesventura commented 5 years ago

You only need to use the options -ngpus and -num_workers to specify the number of GPUs and workers you want to use. This allows you to increase the batch size, so you should also change the option -batch_size to increase it.