def mini_batch_size_update(mini_batch_size, gpu_count):
    """Increase mini-batch size to accommodate more than 1 GPU."""
    if gpu_count <= 1:
        return mini_batch_size
    elif gpu_count > 1:
        return mini_batch_size * gpu_count