cerebras.modelzoo.losses.GPTLMHeadModelLoss.GPTLMHeadModelLoss

class cerebras.modelzoo.losses.GPTLMHeadModelLoss.GPTLMHeadModelLoss(*args, **kwargs)

Bases: torch.nn.Module

Methods

forward

previous

cerebras.modelzoo.losses.GPTLMHeadModelLoss

next

cerebras.modelzoo.losses.LoadBalancingLoss