commit

a0405u · Jan 11, 2021 · 19ddb9a · 19ddb9a
1 parent cf31a69
commit 19ddb9a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/lab3/lab3.py b/lab3/lab3.py
@@ -32,8 +32,8 @@
 
 timest = time.time()
 
-# model = torch.hub.load('pytorch/vision:v0.8.0', 'wide_resnet101_2', pretrained=True).eval().cuda()
-model = alexnet(pretrained = True).eval().cuda()
+model = torch.hub.load('pytorch/vision:v0.8.0', 'wide_resnet101_2', pretrained=True).eval().cuda()
+# model = alexnet(pretrained = True).eval().cuda()
 
 print("model loaded in {}s".format(round(time.time() - timest, 3)))