refactor: remove redundant code

WanglifuCV · Aug 8, 2020 · ae13c5e · ae13c5e
1 parent 65dcbcb
commit ae13c5e
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/test.py b/test.py
@@ -42,9 +42,6 @@
     print(net)
     net.eval()
 
-    if args.gpu:
-        net = net.cuda()
-
     correct_1 = 0.0
     correct_5 = 0.0
     total = 0