Skip to content

Commit c172618

Browse files
committedJun 14, 2023
Replace efficientnet
Loading efficientnet did not work as it is not included in the hubconf.py. The corresponding github issue can be found here: pytorch/hub#236
1 parent 75e1621 commit c172618

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed
 

Diff for: ‎system-config.yaml

+3-3
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,10 @@ models:
3131
torch.hub.load:
3232
repo: pytorch/vision:v0.10.0
3333
model: resnet50
34-
nvidia_efficientnet_b0:
34+
mobilenet_v3_large:
3535
torch.hub.load:
36-
repo: NVIDIA/DeepLearningExamples:torchhub
37-
model: nvidia_efficientnet_b0
36+
repo: pytorch/vision:v0.10.0
37+
model: mobilenet_v3_large
3838
deit-base:
3939
torch.hub.load:
4040
repo: facebookresearch/deit:main

0 commit comments

Comments
 (0)