CGNet
CGNet copied to clipboard
Set the gpu_nums even if the device_count is 1 or less
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.