kaira.losses.BaseLoss

Inheritance diagram for BaseLoss

class kaira.losses.BaseLoss(*args, **kwargs)

Bases: Module, ABC

Base class for all Kaira loss functions.

This abstract class defines the interface that all loss functions in Kaira must implement. It inherits from nn.Module to ensure compatibility with PyTorch’s training pipeline.