add option mini_batch_size)