scglue.models.scglue.IndSCGLUETrainer.compute_losses

IndSCGLUETrainer.compute_losses(data, epoch)

Compute loss functions

Parameters

• 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

• epoch (int) – Current epoch number

• dsc_only – Whether to compute the discriminator loss only

Return type

Mapping[str, Tensor]

Returns

loss_dict – Dict containing loss values