patching convnext model

infer.py

@@ -32,7 +32,11 @@ def _get_feature_dim(model, arch: str) -> int:
 
 def _attach_head(model, arch: str, head: nn.Module) -> nn.Module:
     if arch.startswith("resnet"):            model.fc = head
-    elif arch.startswith("convnext"):        model.classifier = nn.Sequential(nn.Flatten(1), head)
+    elif arch.startswith("convnext"):
+        ln   = model.classifier[0]   # LayerNorm
+        flat = model.classifier[1]   # Flatten(1)
+        model.classifier = nn.Sequential(ln, flat, head)
+        return model
     elif arch.startswith("efficientnet_v2"): model.classifier = nn.Sequential(nn.Dropout(p=0.0), head)
     elif arch.startswith("mobilenet_v3"):    model.classifier = head
     else: raise ValueError(f"Unsupported arch: {arch}")

models/convNext_tiny_v1/model_config.json

@@ -46,7 +46,7 @@
       256
     ],
     "head_dropout": 0.25,
-    "head_norm": "batchnorm"
+    "head_norm": "layernorm"
   },
   "stage_A": {
     "epochs": 5,

models/manifest.json

@@ -1,4 +1,12 @@
 {
-  "resnet50_v1": { "repo_id": "AAI3001-P20/streetview-resnet50", "filename": "best.pth" },
-  "convNext_tiny_v1": { "repo_id": "AAI3001-P20/streetview-convnext-tiny", "filename": "best.pth" }
+  "resnet50_v1": { 
+    "_comment": "Using 20250930_193300__resnet50__exp02",
+    "repo_id": "AAI3001-P20/streetview-resnet50", 
+    "filename": "best.pth" 
+  }, 
+  "convNext_tiny_v1": { 
+    "_comment": "Using 20251006_230926__convnext_tiny__exp02",
+    "repo_id": "AAI3001-P20/streetview-convnext-tiny", 
+    "filename": "best.pth" 
+  }
 }