I have 4 GPUs, when I run train.py with --num_samples 1 --gpu 4，there is only one GPU has utilization.
Is it because the model does not support multiple GPUs?
But when I run search.py with --num_samples 16 --gpu 0.25 , all GPUs has utilization.