played around with CASIA-WEB-FACE

2026-05-09 23:01:15 +02:00
parent a4191fa00c
commit 770b7be936
6 changed files with 83 additions and 18 deletions
--- a/architectures/ResNet18.py
+++ b/architectures/ResNet18.py
@@ -11,12 +11,12 @@ class ResNet18(Model):
        m = models.resnet18(weights=models.ResNet18_Weights.DEFAULT)

        # freez all layers
-        for param in m.parameters():
-            param.requires_grad = False
+        #for param in m.parameters():
+        #    param.requires_grad = False

        # unfreez the last two
-        for param in m.layer3.parameters(): param.requires_grad = True
-        for param in m.layer4.parameters(): param.requires_grad = True
+        #for param in m.layer3.parameters(): param.requires_grad = True
+        #for param in m.layer4.parameters(): param.requires_grad = True
        
        m.fc = nn.Linear(m.fc.in_features, self.size)
        return m