We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f61dde4 commit 306e762Copy full SHA for 306e762
simclr/modules/resnet.py
@@ -3,8 +3,8 @@
3
4
def get_resnet(name, pretrained=False):
5
resnets = {
6
- "resnet18": torchvision.models.resnet18(pretrained=pretrained),
7
- "resnet50": torchvision.models.resnet50(pretrained=pretrained),
+ "resnet18": torchvision.models.resnet18(weights=torchvision.models.ResNet18_Weights.DEFAULT),
+ "resnet50": torchvision.models.resnet50(weights=torchvision.models.ResNet50_Weights.DEFAULT),
8
}
9
if name not in resnets.keys():
10
raise KeyError(f"{name} is not a valid ResNet version")
0 commit comments