scglue.models.glue.GLUETrainer.compute_losses

GLUETrainer.compute_losses(data, epoch, dsc_only=False)

Compute loss functions

参数

• data (typing.Tuple[typing.Mapping[str, torch.Tensor], typing.Mapping[str, torch.Tensor], torch.Tensor, torch.Tensor, torch.Tensor]) – Data tensors

• epoch (int) – Current epoch number

• dsc_only (bool) – Whether to compute the discriminator loss only

返回类型

typing.Mapping[str, torch.Tensor]

返回

loss_dict – Dict containing loss values