modified resnet_model.py

models/resnet_model.py

@@ -7,7 +7,7 @@ class MonkeyResNet(nn.Module):
         super(MonkeyResNet, self).__init__()
         
         # Load pretrained ResNet18 model from torchvision
-        self.model = models.resnet18(pretrained=True)  # pretrained weights from ImageNet
+        self.model = models.resnet18(pretrained=False) 
 
         # Replace the final fully connected layer to match number of classes
         in_features = self.model.fc.in_features