Spaces:
Sleeping
Sleeping
Commit ·
834bff8
1
Parent(s): 44be7f5
Fix bug in down sample
Browse files
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().
|
| 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)
|