Source code for espnet2.enh.loss.criterions.abs_loss

from abc import ABC, abstractmethod

import torch

EPS = torch.finfo(torch.get_default_dtype()).eps

[docs]class AbsEnhLoss(torch.nn.Module, ABC): """Base class for all Enhancement loss modules.""" # the name will be the key that appears in the reporter @property def name(self) -> str: return NotImplementedError # This property specifies whether the criterion will only # be evaluated during the inference stage @property def only_for_test(self) -> bool: return False
[docs] @abstractmethod def forward( self, ref, inf, ) -> torch.Tensor: # the return tensor should be shape of (batch) raise NotImplementedError