ImportError: cannot import name 'save_config'

❓ Questions and Help

Hello

I would like to a multigpu training. I check if my version of pytorch supported the multigpu training thanks to the command: torch.distributed.is_available() Its the case. But when I try: python -m torch.distributed.launch --nproc_per_node=2 tools/train_net.py --config-file "configs/dcn/e2e_mask_rcnn_dconv_R_50_FPN_1x.yaml" I receive the following error:

Traceback (most recent call last):
  File "tools/train_net.py", line 26, in <module>
    from maskrcnn_benchmark.utils.miscellaneous import mkdir, save_config
ImportError: cannot import name 'save_config'

The thing is I arrive to call save_config from the interpreter. Any idea how I could make it works for multi-gpu.

[jaemin93]

I know why this problem occurred. install seems to be working, but it is not. Make sure to place the maskrcnn packages properly in your Python environment.

1. Install $ python setup.py build develop

2. Find maskrcnn packages location $ find / -name 'miscellaneous.py'

/root/maskrcnn-benchmark/build/lib.linux-x86_64-3.6/maskrcnn_benchmark/utils/miscellaneous.py
/root/maskrcnn-benchmark/maskrcnn_benchmark/utils/miscellaneous.py
/opt/conda/lib/python3.6/site-packages/maskrcnn_benchmark/utils/miscellaneous.py
/opt/pytorch/examples/maskrcnn/pytorch/maskrcnn_benchmark/utils/miscellaneous.py
/workspace/examples/maskrcnn/pytorch/maskrcnn_benchmark/utils/miscellaneous.py

1. Remove and Copy

$ rm -rf /opt/conda/lib/python3.6/site-packages/maskrcnn_benchmark
$ cp -r maskrcnn-benchmark/build/lib.linux-x86_64-3.6/maskrcnn_benchmark /opt/conda/lib/python3.6/site-packages/

It worked in my case.

[wanghao14]

@jaemin93 Thanks a lot! Your solution works for me, too.