added more models for testing

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)
+        return m