scglue.models.base.Trainer.get_losses

Trainer.get_losses(loader)[source]

Get loss values for given data

Parameters

loader (Iterable) – Data loader

Return type

Mapping[str, float]

Returns

loss_dict – Dict containing loss values