Skip to content

Commit d17b4de

Browse files
committed
Update ResNet model to use models.resnet50
1 parent baaa499 commit d17b4de

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tools/perf/hub.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
BENCHMARK_MODELS = {
2929
"vgg16": {"model": models.vgg16(pretrained=True), "path": ["script", "pytorch"]},
3030
"resnet50": {
31-
"model": torch.hub.load("pytorch/vision:v0.9.0", "resnet50", pretrained=True),
31+
"model": models.resnet50(weights=None),
3232
"path": ["script", "pytorch"],
3333
},
3434
"efficientnet_b0": {

tools/perf/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
BENCHMARK_MODELS = {
88
"vgg16": {"model": models.vgg16(pretrained=True), "path": ["script", "pytorch"]},
99
"resnet50": {
10-
"model": torch.hub.load("pytorch/vision:v0.9.0", "resnet50", pretrained=True),
10+
"model": models.resnet50(weights=None),
1111
"path": ["script", "pytorch"],
1212
},
1313
"efficientnet_b0": {

0 commit comments

Comments
 (0)