tomofi commited on
Commit
3ecd77f
1 Parent(s): c805e38

Update maskrcnn_benchmark/utils/model_zoo.py

Browse files
maskrcnn_benchmark/utils/model_zoo.py CHANGED
@@ -7,7 +7,10 @@ try:
7
  from torch.hub import urlparse
8
  from torch.hub import HASH_REGEX
9
  except ImportError:
10
- from torch.utils.model_zoo import _download_url_to_file
 
 
 
11
  from torch.utils.model_zoo import urlparse
12
  from torch.utils.model_zoo import HASH_REGEX
13
 
 
7
  from torch.hub import urlparse
8
  from torch.hub import HASH_REGEX
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
  from torch.utils.model_zoo import urlparse
15
  from torch.utils.model_zoo import HASH_REGEX
16