Automatically reduce batch size for small validation/test splits

Thanks @antihutka