Flag not recognized: No module named --batch_size

Have a look at the flag value decoding documentation and arg-split.

Note that guild has integrated support for choosing GPUs.