From 96c6108ec6994998405b51e57dccca8c6cbb3bf5 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 5 Oct 2022 11:49:27 +0530 Subject: [PATCH] document average loss --- enhancer/loss.py | 2 ++ 1 file changed, 2 insertions(+) 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):