Remove batch size setting