mindformers.models.gpt2.GPT2LMHeadModel

class mindformers.models.gpt2.GPT2LMHeadModel(config: mindformers.models.gpt2.gpt2_config.GPT2Config = None)

Provide gpt training loss or logits through network. :param config: The config of Gpt2Model. :type config: GPT2Config

返回

Tensor, the loss or logits of the network.