All models support aux_params parameters, which is default set toNone. If aux_params = None than classification auxiliary outputis not created, else model produce not only mask, but also labeloutput with shape NC. Classification head consist ofGlobalPooling->Dropout(optional)->Linear->Activation(optional) layers,which can be configured by aux_params as follows: