batch_size tweak