Update model.py

model.py

@@ -10,7 +10,7 @@ def create_model(num_of_classes:int=3):
         parameter.requires_grad=False
     for parameter in model.classifier[-4:].parameters():
         parameter.requires_grad=True
-    for parameter in model.features[-2:].parameters():
+    for parameter in model.features[-3:].parameters():
         parameter.requires_grad=True
     model.classifier[3]=nn.Sequential(nn.Linear(1280,1000),nn.ReLU(),nn.Dropout(p=0.3),nn.Linear(1000,num_of_classes))
     return model,transform