changed to not download pretrained weights for ResNet18

2024-07-28 21:01:19 +02:00
parent 6c354b23bd
commit e6fdbfe0b0
1 changed files with 2 additions and 2 deletions
@@ -9,7 +9,7 @@ class ResNet18Head(nn.Module):
        super().__init__()
        # copy out parts from ResNet18 with weights
        resnet18 = torchvision.models.resnet18(
-            weights=torchvision.models.ResNet18_Weights.IMAGENET1K_V1,
+            # weights=torchvision.models.ResNet18_Weights.IMAGENET1K_V1,
        )
        # save relevant layers
        self.conv1 = resnet18.conv1