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