custom batch size
custom batch size