Add cores usage and use 100*precision

juexinwang · May 24, 2020 · 16350af · 16350af
1 parent 91a65be
commit 16350af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main_benchmark.py b/main_benchmark.py
@@ -155,7 +155,7 @@
 torch.manual_seed(args.seed)
 device = torch.device("cuda" if args.cuda else "cpu")
 
-if not args.coresUsage == 'all'
+if not args.coresUsage == 'all':
     torch.set_num_threads(int(args.coresUsage))
 
 kwargs = {'num_workers': 1, 'pin_memory': True} if args.cuda else {}