Upload config.yaml

config.yaml

@@ -0,0 +1,171 @@
+# Kolam AI Generator Configuration
+
+# Data Configuration
+data:
+  input_channels: 1  # Grayscale images
+  image_size: 64     # Target image size
+  batch_size: 32
+  num_workers: 4
+  train_ratio: 0.8
+  val_ratio: 0.1
+  test_ratio: 0.1
+
+# Model Configuration
+models:
+  # CNN Feature Extractor
+  cnn:
+    feature_dim: 128
+    num_classes: 10
+    learning_rate: 0.001
+    weight_decay: 1e-4
+    lr_step_size: 30
+    lr_gamma: 0.1
+  
+  # GAN Configuration
+  gan:
+    noise_dim: 100
+    feature_dim: 128
+    output_channels: 1
+    image_size: 64
+    use_spectral_norm: true
+    
+    # Generator
+    generator:
+      learning_rate: 0.0002
+      beta1: 0.5
+      beta2: 0.999
+    
+    # Discriminator
+    discriminator:
+      learning_rate: 0.0002
+      beta1: 0.5
+      beta2: 0.999
+      n_critic: 1  # Number of discriminator updates per generator update
+
+# Training Configuration
+training:
+  epochs: 200
+  save_interval: 50
+  sample_interval: 10
+  checkpoint_dir: "models/checkpoints"
+  
+  # Early stopping
+  early_stopping:
+    patience: 20
+    min_delta: 0.001
+  
+  # Gradient clipping
+  gradient_clipping:
+    enabled: true
+    max_norm: 1.0
+
+# Data Augmentation
+augmentation:
+  enabled: true
+  transforms:
+    - "random_horizontal_flip"
+    - "random_rotation"
+    - "random_brightness"
+    - "random_contrast"
+  
+  # Parameters
+  rotation_degrees: 15
+  brightness_factor: 0.2
+  contrast_factor: 0.2
+
+# Evaluation Configuration
+evaluation:
+  num_samples: 1000
+  metrics:
+    - "inception_score"
+    - "fid_score"
+    - "diversity"
+    - "sharpness"
+    - "symmetry"
+    - "complexity"
+    - "balance"
+    - "rhythm"
+
+# Generation Configuration
+generation:
+  num_samples: 16
+  save_samples: true
+  sample_format: "png"
+  sample_quality: 95
+  
+  # Variations
+  variations:
+    geometric_transforms:
+      - "flip_h"
+      - "flip_v"
+      - "rotate_90"
+      - "rotate_180"
+      - "rotate_270"
+    
+    style_variations:
+      - "enhance_contrast"
+      - "enhance_brightness"
+      - "enhance_sharpness"
+    
+    color_variations:
+      - "grayscale"
+      - "inverted"
+      - "sepia"
+
+# Paths Configuration
+paths:
+  data_dir: "data"
+  raw_data_dir: "data/raw"
+  processed_data_dir: "data/processed"
+  generated_data_dir: "data/generated"
+  model_dir: "models"
+  checkpoint_dir: "models/checkpoints"
+  output_dir: "outputs"
+  logs_dir: "logs"
+
+# Logging Configuration
+logging:
+  level: "INFO"
+  format: "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+  file: "logs/training.log"
+  console: true
+
+# Hardware Configuration
+hardware:
+  device: "auto"  # auto, cpu, cuda
+  mixed_precision: false
+  num_gpus: 1
+
+# Reproducibility
+reproducibility:
+  seed: 42
+  deterministic: true
+  benchmark: false
+
+# Advanced Configuration
+advanced:
+  # Feature extraction
+  feature_extraction:
+    pretrained: false
+    freeze_backbone: false
+  
+  # GAN training
+  gan_training:
+    label_smoothing: 0.1
+    feature_matching: false
+    gradient_penalty: false
+    penalty_weight: 10.0
+  
+  # Model architecture
+  architecture:
+    # Generator
+    generator:
+      use_self_attention: false
+      use_spectral_norm: true
+      dropout_rate: 0.0
+    
+    # Discriminator
+    discriminator:
+      use_self_attention: false
+      use_spectral_norm: true
+      dropout_rate: 0.0