fix cpu/gpu switch

[sam-cal]

args.gpus="0" was considered as a string and not as an integer, resulting to the use of GPU

[hqucms]

Hi @sam-cal -- the convention here is --gpus [GPU IDs], so 0 means to use the GPU device 0. To use CPUs, just set an empty string, i.e., --gpus ''.