kosmos.ml.config.factories.loss

Classes

class LossConfig

Bases: abc.ABC

Loss function configuration for training.

Methods

get_instance() → torch.nn.Module

Get the loss module instance.

Returns:

Loss module instance.

Return type:

Module

---

class CrossEntropyLossConfig

Bases: LossConfig

Cross-entropy loss function configuration.

Methods

get_instance() → torch.nn.CrossEntropyLoss

Get the cross-entropy loss instance.

Returns:

Cross-entropy loss instance.

Return type:

CrossEntropyLoss