scglue.models.base.Trainer.get_losses

Trainer.get_losses(loader)[源代码]

Get loss values for given data

参数:

loader (typing.Iterable) – Data loader

返回类型:

typing.Mapping[str, float]

返回:

loss_dict – Dict containing loss values