Use a smaller batch size