Skip to content

M
maskrcnn
Commit a44d65dc authored by qianyizhang's avatar qianyizhang Committed by Francisco Massa
Browse files
Options

fix imports for latest pytorch-nightly (#698)

parent 08fcf12f
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 3 deletions
maskrcnn_benchmark/utils/model_zoo.py
View file @ a44d65dc
...@@ -2,9 +2,14 @@ ...@@ -2,9 +2,14 @@
import os import os
import sys import sys
from torch.utils.model_zoo import _download_url_to_file try:
from torch.utils.model_zoo import urlparse from torch.utils.model_zoo import _download_url_to_file
from torch.utils.model_zoo import HASH_REGEX from torch.utils.model_zoo import urlparse
from torch.utils.model_zoo import HASH_REGEX
except:
from torch.hub import _download_url_to_file
from torch.hub import urlparse
from torch.hub import HASH_REGEX
from maskrcnn_benchmark.utils.comm import is_main_process from maskrcnn_benchmark.utils.comm import is_main_process
from maskrcnn_benchmark.utils.comm import synchronize from maskrcnn_benchmark.utils.comm import synchronize
......
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