ksuit.trainers.types ==================== .. py:module:: ksuit.trainers.types Attributes ---------- .. autoapisummary:: ksuit.trainers.types.LossResult Classes ------- .. autoapisummary:: ksuit.trainers.types.TrainerResult Module Contents --------------- .. py:class:: TrainerResult .. py:attribute:: total_loss :type: torch.Tensor .. py:attribute:: losses_to_log :type: dict[str, torch.Tensor] | None :value: None .. py:attribute:: additional_outputs :type: dict[str, torch.Tensor] | None :value: None .. py:data:: LossResult