Ensure that the batch_size is not 0