document average loss
This commit is contained in:
parent
756465c2bf
commit
96c6108ec6
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Reference in New Issue