From d0c379167951fc2a3943cae0b0a090eb86077f6a Mon Sep 17 00:00:00 2001 From: Pedro Larroy Date: Wed, 22 May 2019 14:53:28 -0700 Subject: [PATCH] Fix typo --- tools/caffe_converter/test_converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/caffe_converter/test_converter.py b/tools/caffe_converter/test_converter.py index b79e91f729e1..a4f744556fed 100644 --- 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