Upload config.py with huggingface_hub

config.py

@@ -0,0 +1,61 @@
+"""
+Configuration management for Palette-Reasoner
+"""
+
+from dataclasses import dataclass, field
+from typing import Optional
+
+
+@dataclass
+class ModelConfig:
+    """Model hyperparameters"""
+    hidden_dim: int = 768
+    patch_size: int = 4
+    vit_layers: int = 6
+    vit_heads: int = 8
+    trm_halt_threshold: float = 0.95
+    action_space: int = 512
+    palette_size: int = 4096
+
+
+@dataclass
+class TrainingConfig:
+    """Training hyperparameters"""
+    batch_size: int = 8
+    learning_rate: float = 1e-4
+    weight_decay: float = 0.01
+    max_epochs: int = 100
+    gradient_clip_norm: float = 1.0
+    ponder_lambda: float = 0.01
+    halt_entropy_lambda: float = 0.1
+    warmup_steps: int = 100
+    max_steps: int = 10000
+
+
+@dataclass
+class DataConfig:
+    """Dataset configuration"""
+    data_dir: str = "./data/samples"
+    train_split: float = 0.8
+    val_split: float = 0.1
+    test_split: float = 0.1
+    max_palette_height: int = 128
+    max_palette_width: int = 64
+    synthetic_samples: int = 1000
+
+
+@dataclass
+class Config:
+    """Master configuration"""
+    model: ModelConfig = field(default_factory=ModelConfig)
+    training: TrainingConfig = field(default_factory=TrainingConfig)
+    data: DataConfig = field(default_factory=DataConfig)
+    seed: int = 42
+    device: str = "cuda"
+    output_dir: str = "./outputs"
+    log_dir: str = "./logs"
+
+
+def get_default_config() -> Config:
+    """Get default configuration"""
+    return Config()