Changes AutoEncoder Hidden Layers

iSiddharth20 · Jan 2, 2024 · 70742a1 · 70742a1
1 parent 5f28ffa
commit 70742a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Code/autoencoder_model.py b/Code/autoencoder_model.py
@@ -16,9 +16,9 @@ class Grey2RGBAutoEncoder(nn.Module):
     def __init__(self):  
         super(Grey2RGBAutoEncoder, self).__init__()  
         # Define the Encoder
-        self.encoder = self._make_layers([1, 64, 128, 256])
+        self.encoder = self._make_layers([1, 8, 16, 32])
         # Define the Decoder
-        self.decoder = self._make_layers([256, 128, 64, 3], decoder=True)
+        self.decoder = self._make_layers([32, 16, 8, 3], decoder=True)
 
     # Helper function to create the encoder or decoder layers.
     def _make_layers(self, channels, decoder=False):

diff --git a/Code/main.py b/Code/main.py
@@ -22,8 +22,8 @@
 rgb_dir = '../Dataset/RGB'
 
 # Define Universal Parameters
-image_height = 400
-image_width = 600
+image_height = 3000
+image_width = 4500
 batch_size = 2
 
 def main_worker(rank, world_size):