loss/abstract_loss_func.py

import torch.nn as nn

class AbstractLossClass(nn.Module):
    """Abstract class for loss functions."""
    def __init__(self):
        super(AbstractLossClass, self).__init__()

    def forward(self, pred, label):
        """
        Args:
            pred: prediction of the model
            label: ground truth label
            
        Return:
            loss: loss value
        """
        raise NotImplementedError('Each subclass should implement the forward method.')