Fix typo

apache · May 22, 2019 · d0c3791 · d0c3791
1 parent 3615916
commit d0c3791
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/caffe_converter/test_converter.py b/tools/caffe_converter/test_converter.py
@@ -90,7 +90,7 @@ def main():
         gpus = [-1]
         default_batch_size = 32
     else:
-        num_gpus = mx.context.nu_gpus()
+        num_gpus = mx.context.num_gpus()
         assert num_gpus, 'At least one GPU is needed to run test_converter in GPU mode'
         default_batch_size = 32 * num_gpus