cerebras.modelzoo.losses.GPTLMHeadModelLoss.GPTLMHeadModelLoss#

class cerebras.modelzoo.losses.GPTLMHeadModelLoss.GPTLMHeadModelLoss(*args, **kwargs)[source]#

Bases: torch.nn.Module

Methods

forward