search

wutianyiRosun / CGNet

CGNet: A Light-weight Context Guided Network for Semantic Segmentation [IEEE Transactions on Image Processing 2020]
MIT License
259 stars 51 forks source link

Set the gpu_nums even if the device_count is 1 or less #28

Open sabughazal opened 2 years ago

sabughazal commented 2 years ago

Set the gpu_nums even if the device_count is 1 or less. This line has to be there when training on a single gpu.