diff options
| author | rohithkrn <rohith.nallamaddi@gmail.com> | 2019-01-22 17:19:15 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-22 17:28:04 -0800 |
| commit | ddeaa541aa5ff5c13c8fadf43a15eb142c3be207 (patch) | |
| tree | 7b4d8ef37c9a74f2a977a3cd8f3f627b4b950027 | |
| parent | e7a77ac3b0d76254ea6413021d7eec369d9c1ac9 (diff) | |
| download | pytorch-ddeaa541aa5ff5c13c8fadf43a15eb142c3be207.tar.gz pytorch-ddeaa541aa5ff5c13c8fadf43a15eb142c3be207.tar.bz2 pytorch-ddeaa541aa5ff5c13c8fadf43a15eb142c3be207.zip | |
fix typo in resnet50_trainer.py
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/16219
Differential Revision: D13776742
Pulled By: bddppq
fbshipit-source-id: 10a6ab4c58159b3f619b739074f773662722c1d9
| -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, |