Spaces:
Running
Running
File size: 687 Bytes
aead140 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
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!")
|