updated model for learned segmenter