diff --git a/enhancer/loss.py b/enhancer/loss.py index 1e156a0..f2f62d3 100644 --- a/enhancer/loss.py +++ b/enhancer/loss.py @@ -94,6 +94,8 @@ class Avergeloss(nn.Module): """ Combine multiple metics of same nature. for example, ["mea","mae"] + parameters: + losses : loss function names to be combined """ def __init__(self, losses):