adityachaubey commited on
Commit
834bff8
·
1 Parent(s): 44be7f5

Fix bug in down sample

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -67,7 +67,7 @@ class resblock(torch.nn.Module):
67
  #Downsmaple and upsmaple
68
  class down_sample(torch.nn.Module):
69
  def __init__(self, channels):
70
- super().__ini__()
71
  self.conv = nn.Conv2d(channels, channels, 4, stride=2, padding=1)
72
  def forward(self, x):
73
  return self.conv(x)
 
67
  #Downsmaple and upsmaple
68
  class down_sample(torch.nn.Module):
69
  def __init__(self, channels):
70
+ super().__init__()
71
  self.conv = nn.Conv2d(channels, channels, 4, stride=2, padding=1)
72
  def forward(self, x):
73
  return self.conv(x)