fix: use usize instead of u64 for batch-size arg

fix: use usize instead of u64 for batch-size arg #92