Source code for mridc.core.classes.loss

# encoding: utf-8
__author__ = "Dimitrios Karkalousos"

# Taken and adapted from: https://github.com/NVIDIA/NeMo/blob/main/nemo/core/classes/loss.py

import torch

__all__ = ["Loss"]

from mridc.core.classes.common import Serialization, Typing


[docs]class Loss(torch.nn.modules.loss._Loss, Typing, Serialization): """Inherit this class to implement custom loss."""