ksuit.trainers.types

Attributes

LossResult

Classes

TrainerResult

Module Contents

class ksuit.trainers.types.TrainerResult

total_loss: torch.Tensor

losses_to_log: dict[str, torch.Tensor] | None = None

additional_outputs: dict[str, torch.Tensor] | None = None

ksuit.trainers.types.LossResult