Spaces:
Runtime error
Runtime error
Update maskrcnn_benchmark/utils/model_zoo.py
Browse files
maskrcnn_benchmark/utils/model_zoo.py
CHANGED
|
@@ -9,10 +9,12 @@ try:
|
|
| 9 |
except ImportError:
|
| 10 |
try:
|
| 11 |
from torch.utils.model_zoo import _download_url_to_file
|
|
|
|
|
|
|
| 12 |
except ImportError:
|
| 13 |
from torch.hub import download_url_to_file as _download_url_to_file
|
| 14 |
-
|
| 15 |
-
|
| 16 |
|
| 17 |
from maskrcnn_benchmark.utils.comm import is_main_process
|
| 18 |
from maskrcnn_benchmark.utils.comm import synchronize
|
|
|
|
| 9 |
except ImportError:
|
| 10 |
try:
|
| 11 |
from torch.utils.model_zoo import _download_url_to_file
|
| 12 |
+
from torch.utils.model_zoo import urlparse
|
| 13 |
+
from torch.utils.model_zoo import HASH_REGEX
|
| 14 |
except ImportError:
|
| 15 |
from torch.hub import download_url_to_file as _download_url_to_file
|
| 16 |
+
from torch.hub import urlparse
|
| 17 |
+
from torch.hub import HASH_REGEX
|
| 18 |
|
| 19 |
from maskrcnn_benchmark.utils.comm import is_main_process
|
| 20 |
from maskrcnn_benchmark.utils.comm import synchronize
|