increase the batch size at runtime, given GPU memory is big enough
increase the batch size at runtime, given GPU memory is big enough