commit

a0405u · Dec 29, 2020 · 118a248 · 118a248
1 parent d23596f
commit 118a248
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lab3/lab3.py b/lab3/lab3.py
@@ -31,7 +31,7 @@
 
 timest = time.time()
 
-model = wide_resnet101_2(pretrained = True).eval().cuda()
+model = torch.hub.load('pytorch/vision:v0.8.0', 'wide_resnet101_2', pretrained=True)
 # model = alexnet(pretrained = True).eval().cuda()
 
 print("model loaded in {}s".format(round(time.time() - timest, 2)))