`kosmos.ml.config.factories.loss`¶

Classes¶

class LossConfig[source]¶

Bases: abc.ABC

Loss function configuration for training.

Methods

get_instance() → torch.nn.Module[source]¶

Get the loss module instance.

Returns:: Loss module instance.
Return type:: Module

class CrossEntropyLossConfig[source]¶

Bases: LossConfig

Cross-entropy loss function configuration.

Methods

get_instance() → torch.nn.CrossEntropyLoss[source]¶

Get the cross-entropy loss instance.

Returns:: Cross-entropy loss instance.
Return type:: CrossEntropyLoss