noether.training.trainers.types

Attributes

LossResult

Classes

TrainerResult

Module Contents

class noether.training.trainers.types.TrainerResult

total_loss: torch.Tensor

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

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

noether.training.trainers.types.LossResult