twodgirl commited on
Commit
e9f21ca
1 Parent(s): d213e90
Files changed (1) hide show
  1. 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 TeaEncoder(nn.Module):
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(