Method head

modelscope/models/nlp/task_models/task_model.py:576–579

(self)

    @property
    def head(self):
        if 'head' != self.head_prefix:
            return getattr(self, self.head_prefix)
        return super().__getattr__('head')

    def extract_feature(self, **input: Dict[str, Any]) -> Dict[str, Any]:
        """default forward method is the backbone-only forward"""