Change batch size to 100 to get better perf: