How to change number of gpus

nv-tlabs / GSCNN

Gated-Shape CNN for Semantic Segmentation (ICCV 2019)

https://nv-tlabs.github.io/GSCNN/

Other

915 stars 200 forks source link

How to change number of gpus #25

Closed shubhaminnani closed 4 years ago

shubhaminnani commented 4 years ago

Hi, i want to know how to use multiple gpu for training with multiple batch size .. and validate with single batch size @tovacinni @varunjampani @davidjesusacu @ShreyasSkandanS

tovacinni commented 4 years ago

Not sure what you mean exactly, but take a look at the parameters bs_mult and bs_mult_val.

shubhaminnani commented 4 years ago

thanks, will check this out