update batch size for training