Fix web UI initialization, enforce strict config validation, and align trainer/orchestrator behavior

config/inference.yaml

@@ -2,54 +2,57 @@ seed: 1
 device: cuda
 
 data:
-  root: data/mini_deepfake_dataset
+  root: data/deepfake_dataset_sample
   test_split: test
-  val_split: val
   num_classes: 2
   img_size: 224
 
+defaults:
+  model:
+    kind: timm
+    pretrained: false
+  training:
+    batch_size: 64
+    num_workers: 4
+  inference:
+    batch_size: 64
+    num_workers: 4
+  transforms:
+    eval:
+      resize: true
+      center_crop: true
+
 models:
-  efficientnet_b3:
-    output_dir: runs/efficientnet_b3
-    transforms:
-      eval:
-        ensure_rgb: true
-        val_resize: true
-        val_center_crop: true
+  repvit_m1:
+    model:
+      id: repvit_m1.dist_in1k
     inference:
-      weights: weights/efficientnet_b3.pth
-      split: test
-      batch_size: 256
-      num_workers: 2
-      img_size: 224
+      weights: weights/repvit_m1.pth
+
   efficientformerv2_s1:
-    output_dir: runs/efficientformerv2_s1
-    transforms:
-      eval:
-        ensure_rgb: true
-        val_resize: true
-        val_center_crop: true
+    model:
+      id: efficientformerv2_s1
     inference:
       weights: weights/efficientformerv2_s1.pth
-      split: test
-      batch_size: 256
-      num_workers: 4
-      img_size: 224
+
   faster_vit_2_224:
-    output_dir: runs/faster_vit_2_224
-    transforms:
-      eval:
-        ensure_rgb: true
-        val_resize: true
-        val_center_crop: true
+    model:
+      kind: import
+      builder: orchestration.model_registry:build_fastervit
+      kwargs:
+        model_name: faster_vit_2_224
+        pretrained: false
     inference:
       weights: weights/faster_vit_2_224.pth
-      split: test
-      batch_size: 256
-      num_workers: 4
-      img_size: 224
+
+  efficientnet_b3:
+    model:
+      id: efficientnet_b3
+    inference:
+      weights: weights/efficientnet_b3.pth
 
 selection:
-  - efficientnet_b3
+  - repvit_m1
   - efficientformerv2_s1
   - faster_vit_2_224
+  - efficientnet_b3

config/train.yaml

@@ -1,69 +1,72 @@
+# Global experiment settings
 seed: 1
 device: cuda
 
 data:
-  root: data/mini_deepfake_dataset
+  root: data/deepfake_dataset_sample
   train_split: train
   val_split: val
+  test_split: test
   num_classes: 2
   img_size: 224
 
+defaults:
+  model:
+    kind: timm
+    pretrained: true
+  training:
+    epochs: 5
+    batch_size: 64
+    num_workers: 8
+    lr: 1.0e-4
+    weight_decay: 5.0e-2
+    resume: auto
+  inference:
+    batch_size: 64
+    num_workers: 4
+  transforms:
+    train:
+      random_resized_crop: true
+      random_flip: true
+      color_jitter: true
+    eval:
+      resize: true
+      center_crop: true
+
+# Per-model configuration
 models:
-  efficientnet_b3:
-    output_dir: runs/efficientnet_b3
-    transforms:
-      train:
-        ensure_rgb: true
-        train_random_resized_crop: true
-        train_random_horizontal_flip: true
-        train_random_rotation: true
-        train_color_jitter: true
-        train_random_erasing: true
-      eval:
-        ensure_rgb: true
-        val_resize: true
-        val_center_crop: true
+  repvit_m1:
+    model:
+      id: repvit_m1.dist_in1k
     training:
-      epochs: 5
-      batch_size: 64
-      num_workers: 4
-      resume: auto
+      epochs: 8
+
   efficientformerv2_s1:
-    output_dir: runs/efficientformerv2_s1
-    transforms:
-      train:
-        ensure_rgb: true
-        train_random_resized_crop: true
-        train_random_horizontal_flip: true
-        train_color_jitter: true
-      eval:
-        ensure_rgb: true
-        val_resize: true
-        val_center_crop: true
+    model:
+      id: efficientformerv2_s1
     training:
-      epochs: 5
-      batch_size: 128
-      num_workers: 8
-      resume: auto
+      epochs: 6
+
   faster_vit_2_224:
-    output_dir: runs/faster_vit_2_224
-    transforms:
-      train:
-        ensure_rgb: true
-        train_random_resized_crop: true
-        train_random_horizontal_flip: true
-        train_color_jitter: true
-      eval:
-        ensure_rgb: true
-        val_resize: true
-        val_center_crop: true
+    model:
+      kind: import
+      builder: orchestration.model_registry:build_fastervit
+      kwargs:
+        model_name: faster_vit_2_224
+        pretrained: true
     training:
-      epochs: 5
-      batch_size: 64
-      num_workers: 8
-      resume: auto
+      epochs: 6
+      batch_size: 48
 
+  efficientnet_b3:
+    model:
+      id: efficientnet_b3
+    training:
+      epochs: 8
+
+# Which models to run
 selection:
+  - faster_vit_2_224
+  - repvit_m1
   - efficientnet_b3
   - efficientformerv2_s1
-  - faster_vit_2_224