Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

document average loss

This commit is contained in:
shahules786 2022-10-05 11:49:27 +05:30
parent 756465c2bf
commit 96c6108ec6
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/loss.py
View File

@ -94,6 +94,8 @@ class Avergeloss(nn.Module):
""" """
Combine multiple metics of same nature. Combine multiple metics of same nature.
for example, ["mea","mae"] for example, ["mea","mae"]
parameters:
losses : loss function names to be combined
""" """
def __init__(self, losses): def __init__(self, losses):