From 6ee43c386eb9aa08524bf3a2b900af88017bd09b Mon Sep 17 00:00:00 2001 From: Shu Peng Date: Sun, 14 Apr 2019 18:46:01 -0700 Subject: [PATCH] APIs moved to torch.hub --- maskrcnn_benchmark/utils/model_zoo.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/maskrcnn_benchmark/utils/model_zoo.py b/maskrcnn_benchmark/utils/model_zoo.py index 7a0ebb349..61facb82a 100644 --- a/maskrcnn_benchmark/utils/model_zoo.py +++ b/maskrcnn_benchmark/utils/model_zoo.py @@ -2,9 +2,10 @@ import os import sys -from torch.utils.model_zoo import _download_url_to_file -from torch.utils.model_zoo import urlparse -from torch.utils.model_zoo import HASH_REGEX +# apis moved from torch.utils.model_zoo to torch.hub +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 synchronize