loss.rst

.. module:: tropt.loss

Losses

Loss Resolution

The unified entry point for computing any loss — models call this instead of invoking loss functions directly. See :doc:`common` for ModelInput / ModelOutput.

.. autofunction:: tropt.loss.resolution.resolve_and_compute_loss

.. autoclass:: tropt.loss.resolution.LossResolutionError
   :members:
   :undoc-members:
   :show-inheritance:

---

Loss Classes Interfaces

.. autoclass:: BaseLoss
   :members:
   :undoc-members:
   :show-inheritance:

.. autoclass:: PrefillBasedLoss
   :members:
   :undoc-members:
   :show-inheritance:

.. autoclass:: TextBasedLoss
   :members:
   :undoc-members:
   :show-inheritance:

.. autoclass:: SteeringActivationLoss
   :members:
   :undoc-members:
   :show-inheritance:

.. autoclass:: CombinedLoss
   :members:
   :undoc-members:
   :show-inheritance:

---

Loss Implementations

.. automodule:: tropt.loss
   :members:
   :exclude-members: BaseLoss, PrefillBasedLoss, TextBasedLoss, SteeringActivationLoss, CombinedLoss
   :undoc-members:
   :show-inheritance:
   :imported-members: