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

View File

@ -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):