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