brian/visual_critical_discourse_analysis
1
0
Fork 0
Code Issues 11 Pull Requests Actions Projects 1 Releases Activity

changed to not download pretrained weights for ResNet18

Browse Source
This commit is contained in:
Brian Bjarke Jensen
2024-07-28 21:01:19 +02:00
parent 6c354b23bd
commit e6fdbfe0b0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model/src/models/resnet18_head.py
+1 -1
View File
@@ -9,7 +9,7 @@ class ResNet18Head(nn.Module):
super().__init__() super().__init__()
# copy out parts from ResNet18 with weights # copy out parts from ResNet18 with weights
resnet18 = torchvision.models.resnet18( resnet18 = torchvision.models.resnet18(
weights=torchvision.models.ResNet18_Weights.IMAGENET1K_V1, # weights=torchvision.models.ResNet18_Weights.IMAGENET1K_V1,
) )
# save relevant layers # save relevant layers
self.conv1 = resnet18.conv1 self.conv1 = resnet18.conv1