Set batch size

Prevent use too many memory.