Add downsample and upsample classes for improved model architecture

model.py

@@ -64,3 +64,17 @@ class resblock(torch.nn.Module):
         h = self.silu(h)
         return h + self.residual_conv(x)
 
+#Downsmaple and upsmaple
+class down_sample(torch.nn.Module):
+    def __init__(self, channels):
+        super().__ini__()
+        self.conv = nn.Conv2d(channels, channels, 4, stride=2, padding=1)
+    def forward(self, x):
+        return self.conv(x)
+
+class up_sample(torch.nn.Module):
+    def __init__(self, channels):
+        super().__init__()
+        self.conv = nn.ConvTranspose2d(channels, channels, 4, stride=2, padding=1)
+    def forward(self, x):
+        return self.conv(x)