sc2bench.loss

Functions

extract_org_loss_dict(org_criterion, ...)	Extracts loss(es) from student_outputs inside TrainingBox or DistillationBox in torchdistill.
extract_org_segment_loss(org_criterion, ...)	Computes loss(es) using the original loss module inside TrainingBox or DistillationBox in torchdistill for semantic segmentation models in torchvision.

Classes

BppLoss(entropy_module_path[, reduction])

Bit-per-pixel (or rate) loss.