Typo.
Browse files- tea_model.py +1 -1
tea_model.py
CHANGED
@@ -27,7 +27,7 @@ class Recon(nn.Module):
|
|
27 |
def forward(self, x):
|
28 |
return self.fuse(self.long(x) + self.short(x))
|
29 |
|
30 |
-
class
|
31 |
def __init__(self, ch_in):
|
32 |
super().__init__()
|
33 |
self.block_in = nn.Sequential(
|
|
|
27 |
def forward(self, x):
|
28 |
return self.fuse(self.long(x) + self.short(x))
|
29 |
|
30 |
+
class TeaDecoder(nn.Module):
|
31 |
def __init__(self, ch_in):
|
32 |
super().__init__()
|
33 |
self.block_in = nn.Sequential(
|