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

init super

This commit is contained in:
shahules786 2022-09-12 10:55:52 +05:30
parent 4536a82e90
commit 0e72bb3bb8
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/utils/loss.py
View File

@ -17,6 +17,7 @@ class mean_squared_error(nn.Module):
class mean_absolute_error(nn.Module): class mean_absolute_error(nn.Module):
def __init__(self,reduction="mean"): def __init__(self,reduction="mean"):
super().__init__()
self.loss_fun = nn.L1Loss(reduction=reduction) self.loss_fun = nn.L1Loss(reduction=reduction)
@ -27,6 +28,7 @@ class mean_absolute_error(nn.Module):
class Avergeloss(nn.Module): class Avergeloss(nn.Module):
def __init__(self,losses): def __init__(self,losses):
super().__init__()
self.valid_losses = nn.ModuleList() self.valid_losses = nn.ModuleList()
for loss in losses: for loss in losses: