started unlearning setup

architectures/ResNet50.py

@@ -23,14 +23,14 @@ class ResNet50(Model):
         m = models.resnet50(weights=models.ResNet50_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
         # NOTE: Freezing everything and unfrizing the last 3 yeilded the best performance
-        for param in m.layer2.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
+        #for param in m.layer2.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