Method set_input_embeddings

bert/modeling_utils.py:324–336

Set model's input embeddings Args: value (:obj:`nn.Module`): A module mapping vocabulary to hidden states.

(self, value: nn.Module)

            raise NotImplementedError

    def set_input_embeddings(self, value: nn.Module):
        """
        Set model's input embeddings

        Args:
            value (:obj:`nn.Module`):
                A module mapping vocabulary to hidden states.
        """
        base_model = getattr(self, self.base_model_prefix, self)
        if base_model is not self:
            base_model.set_input_embeddings(value)
        else:
            raise NotImplementedError

    def get_output_embeddings(self):
        """