scglue.models.base.Trainer.get_losses

Trainer.get_losses(loader)[source]

Get loss values for given data

Parameters

loader (typing.Iterable) – Data loader

Return type

typing.Mapping[str, float]

Returns

loss_dict – Dict containing loss values