Commit 7d8da094 authored by shenyunhang's avatar shenyunhang Committed by Facebook Github Bot

Miss a 'None' in return values of create_model() in tools/train_net.py

Summary:
When 'model_final.pkl' exists, the following error occurs:
```shell
INFO train_net.py: 184: model_final.pkl exists; no need to train!
Traceback (most recent call last):
  File "tools/train_net.py", line 280, in <module>
    main()
  File "tools/train_net.py", line 119, in main
    checkpoints = train_model()
  File "tools/train_net.py", line 128, in train_model
    model, weights_file, start_iter, checkpoints, output_dir = create_model()
ValueError: need more than 4 values to unpack
```
It misses a return value.
Closes https://github.com/facebookresearch/Detectron/pull/309

Reviewed By: ir413

Differential Revision: D7366733

Pulled By: rbgirshick

fbshipit-source-id: c37b3b2e5b2aff38d46ca521a9d79613049255cb
parent e7ad1b48
......@@ -182,7 +182,7 @@ def create_model():
final_path = os.path.join(output_dir, 'model_final.pkl')
if os.path.exists(final_path):
logger.info('model_final.pkl exists; no need to train!')
return None, None, {'final': final_path}, output_dir
return None, None, None, {'final': final_path}, output_dir
# Find the most recent checkpoint (highest iteration number)
files = os.listdir(output_dir)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment