Skip to content

Commit da80d25

Browse files
committed
remove __init__ in dump and flatten
1 parent c731d5a commit da80d25

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

src/model_constructor/layers.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,6 @@
2121
class Flatten(nn.Module):
2222
"""flat x to vector"""
2323

24-
def __init__(self):
25-
super().__init__()
26-
2724
def forward(self, x):
2825
return x.view(x.size(0), -1)
2926

@@ -36,9 +33,6 @@ def noop(x):
3633
class Noop(nn.Module):
3734
"""Dummy module"""
3835

39-
def __init__(self):
40-
super().__init__()
41-
4236
def forward(self, x):
4337
return x
4438

0 commit comments

Comments
 (0)