cleaned up for submission

2026-07-10 20:13:13 +02:00
parent 47fbb19761
commit 7eff030e89
38 changed files with 39 additions and 4175 deletions
--- a/architectures/EfficentNet.py
+++ b/architectures/EfficentNet.py
@@ -11,8 +11,8 @@ class EfficientNet(Model):

        m = models.efficientnet_b1(weights=models.EfficientNet_B1_Weights.DEFAULT)

-        # Unfreeze the last block for a lighter touch
-        for param in m.features[-1].parameters(): param.requires_grad = True
+        # Unfreeze the last block 
+       # for param in m.features[-1].parameters(): param.requires_grad = True

        # Standard classifier fix
        m.classifier[1] = nn.Linear(m.classifier[1].in_features, self.size)