Update config.json

config.json

@@ -1,4 +1,33 @@
 {
-  "model": "DCGAN",
-  "torch_dtype": "float16"
-}
+    "model": "GAN",
+    "generator": {
+        "input_size": 100,
+        "layers": [
+            {"type": "ConvTranspose2d", "in_channels": 100, "out_channels": 512, "kernel_size": 4, "stride": 1, "padding": 0, "bias": false},
+            {"type": "BatchNorm2d", "num_features": 512},
+            {"type": "ReLU", "inplace": true},
+
+            {"type": "ConvTranspose2d", "in_channels": 512, "out_channels": 256, "kernel_size": 4, "stride": 2, "padding": 1, "bias": false},
+            {"type": "BatchNorm2d", "num_features": 256},
+            {"type": "ReLU", "inplace": true},
+
+            {"type": "ConvTranspose2d", "in_channels": 256, "out_channels": 128, "kernel_size": 4, "stride": 2, "padding": 1, "bias": false},
+            {"type": "BatchNorm2d", "num_features": 128},
+            {"type": "ReLU", "inplace": true},
+
+            {"type": "ConvTranspose2d", "in_channels": 128, "out_channels": 64, "kernel_size": 4, "stride": 2, "padding": 1, "bias": false},
+            {"type": "BatchNorm2d", "num_features": 64},
+            {"type": "ReLU", "inplace": true},
+
+            {"type": "ConvTranspose2d", "in_channels": 64, "out_channels": 3, "kernel_size": 4, "stride": 2, "padding": 1, "bias": false},
+            {"type": "Tanh"}
+        ]
+    },
+    "discriminator": {
+        // Add discriminator configuration here if needed
+    },
+    "optimizer": {
+        // Add optimizer configuration here if needed
+    },
+    "loss": "BCELoss"
+}