format

configs/base_config.py

@@ -0,0 +1,28 @@
+"""
+Base configuration dictionary for the project.
+Contains default values for common hyperparameters and settings.
+"""
+
+base_config = {
+    # --- Data --- #
+    "data_root": "data/PennFudanPed",  # Default dataset path
+    "output_dir": "outputs",  # Base directory for logs, checkpoints, etc.
+    # --- Hardware --- #
+    "device": "cuda",  # 'cuda' or 'cpu'
+    # --- Model --- #
+    "num_classes": 2,  # Number of classes (including background)
+    # --- Training --- #
+    "batch_size": 2,  # Training batch size
+    "num_epochs": 10,  # Total number of training epochs
+    "seed": 42,  # Random seed for reproducibility
+    # --- Optimizer --- #
+    "lr": 0.005,  # Initial learning rate
+    "momentum": 0.9,
+    "weight_decay": 0.0005,
+    # --- LR Scheduler --- #
+    "lr_step_size": 3,  # Step size for StepLR scheduler
+    "lr_gamma": 0.1,  # Multiplicative factor for StepLR scheduler
+    # --- Logging & Checkpointing --- #
+    "log_freq": 10,  # Log training progress every N batches
+    "checkpoint_freq": 1,  # Save checkpoint every N epochs
+}