Contents:
GLUETrainer.
compute_losses
Compute loss functions
data (Tuple[Mapping[str, Tensor], Mapping[str, Tensor], Tensor, Tensor, Tensor]) – Data tensors
Tuple
Mapping
str
Tensor
epoch (int) – Current epoch number
int
dsc_only (bool) – Whether to compute the discriminator loss only
bool
Mapping[str, Tensor]
loss_dict – Dict containing loss values