mindformers.models.gpt2.GPT2LMHeadModel¶
- class mindformers.models.gpt2.GPT2LMHeadModel(config: Optional[GPT2Config] = None)[源代码]¶
Provide gpt training loss or logits through network. Args:
config (GPT2Config): The config of Gpt2Model.
- Returns:
Tensor, the loss or logits of the network.