diff options
| author | Thomas Dudziak <tomdzk@fb.com> | 2017-06-06 23:59:46 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2017-06-07 00:04:26 -0700 |
| commit | 60c78d6160ca5bcb2f808cf972b2f0f4578d77fb (patch) | |
| tree | 55a4fe56fd2d32c6583175eb09fcbaa5fa086c5b /caffe2/experiments | |
| parent | 4c5d101cafd6e5deff5f5715cc6bf3e549d8cb99 (diff) | |
| download | pytorch-60c78d6160ca5bcb2f808cf972b2f0f4578d77fb.tar.gz pytorch-60c78d6160ca5bcb2f808cf972b2f0f4578d77fb.tar.bz2 pytorch-60c78d6160ca5bcb2f808cf972b2f0f4578d77fb.zip | |
Fixes range/xrange for Python 3
Summary: As title
Differential Revision: D5151894
fbshipit-source-id: 7badce5d3122e8f2526a7170fbdcf0d0b66e2638
Diffstat (limited to 'caffe2/experiments')
| -rw-r--r-- | caffe2/experiments/python/net_construct_bench.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/caffe2/experiments/python/net_construct_bench.py b/caffe2/experiments/python/net_construct_bench.py index 05514a2aab..1ac248392e 100644 --- a/caffe2/experiments/python/net_construct_bench.py +++ b/caffe2/experiments/python/net_construct_bench.py @@ -60,7 +60,7 @@ def AddMomentumParameterUpdate(train_model, LR): def Create(args): - gpus = range(args.num_gpus) + gpus = list(range(args.num_gpus)) log.info("Running on gpus: {}".format(gpus)) # Create CNNModeLhelper object |