diff options
| -rw-r--r-- | caffe2/python/examples/resnet50_trainer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/caffe2/python/examples/resnet50_trainer.py b/caffe2/python/examples/resnet50_trainer.py index 928c85b13b..bab8c33cae 100644 --- a/caffe2/python/examples/resnet50_trainer.py +++ b/caffe2/python/examples/resnet50_trainer.py @@ -497,7 +497,7 @@ def Train(args): test_model = None if (args.test_data is not None): log.info("----- Create test net ----") - if use_ideep: + if args.use_ideep: test_arg_scope = { 'use_cudnn': False, 'cudnn_exhaustive_search': False, |