:py:mod:`kosmos.ml.config.factories.loss` ========================================= .. py:module:: kosmos.ml.config.factories.loss Classes ------- .. py:class:: LossConfig Bases: :py:class:`abc.ABC` Loss function configuration for training. | .. rubric:: Methods .. py:method:: get_instance() -> torch.nn.Module Get the loss module instance. :returns: Loss module instance. :rtype: Module ---- .. py:class:: CrossEntropyLossConfig Bases: :py:class:`LossConfig` Cross-entropy loss function configuration. | .. rubric:: Methods .. py:method:: get_instance() -> torch.nn.CrossEntropyLoss Get the cross-entropy loss instance. :returns: Cross-entropy loss instance. :rtype: CrossEntropyLoss