Create notebooks/kolam_generation_gan.ipynb

notebooks/kolam_generation_gan.ipynb

@@ -0,0 +1,22 @@
+import torch
+import torch.nn as nn
+import torch.optim as optim
+from torch.utils.data import DataLoader
+import torchvision.transforms as transforms
+import numpy as np
+import matplotlib.pyplot as plt
+from pathlib import Path
+import os
+import sys
+
+# Add the project root to Python path for imports
+project_root = Path.cwd().parent
+sys.path.insert(0, str(project_root))
+sys.path.insert(0, str(project_root / 'models'))
+
+# Import our custom GAN models
+# Note: These imports work at runtime even if IDE shows warnings
+from models.gan_generator import KolamGenerator  # type: ignore
+from models.gan_discriminator import KolamDiscriminator  # type: ignore
+
+print("GAN training setup complete!")