init super
This commit is contained in:
parent
4536a82e90
commit
0e72bb3bb8
|
|
@ -17,6 +17,7 @@ class mean_squared_error(nn.Module):
|
|||
class mean_absolute_error(nn.Module):
|
||||
|
||||
def __init__(self,reduction="mean"):
|
||||
super().__init__()
|
||||
|
||||
self.loss_fun = nn.L1Loss(reduction=reduction)
|
||||
|
||||
|
|
@ -27,6 +28,7 @@ class mean_absolute_error(nn.Module):
|
|||
class Avergeloss(nn.Module):
|
||||
|
||||
def __init__(self,losses):
|
||||
super().__init__()
|
||||
|
||||
self.valid_losses = nn.ModuleList()
|
||||
for loss in losses:
|
||||
|
|
|
|||
Loading…
Reference in New Issue