added two modfels

2026-05-03 17:48:51 +02:00
parent a80d996b0c
commit 1c04344ad6
4 changed files with 23 additions and 37 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.to(self.device)