diff --git a/fastreid/modeling/backbones/osnet.py b/fastreid/modeling/backbones/osnet.py index 243c75a..6c55341 100644 --- a/fastreid/modeling/backbones/osnet.py +++ b/fastreid/modeling/backbones/osnet.py @@ -443,6 +443,7 @@ def init_pretrained_weights(model, key=''): cached_file = os.path.join(model_dir, filename) if not os.path.exists(cached_file): + logger.info(f"Pretrain model don't exist, downloading from {model_urls[key]}") if comm.is_main_process(): gdown.download(model_urls[key], cached_file, quiet=False) diff --git a/fastreid/modeling/backbones/resnet.py b/fastreid/modeling/backbones/resnet.py index 89a6464..e58f376 100644 --- a/fastreid/modeling/backbones/resnet.py +++ b/fastreid/modeling/backbones/resnet.py @@ -278,6 +278,7 @@ def init_pretrained_weights(key): cached_file = os.path.join(model_dir, filename) if not os.path.exists(cached_file): + logger.info(f"Pretrain model don't exist, downloading from {model_urls[key]}") if comm.is_main_process(): gdown.download(model_urls[key], cached_file, quiet=False) diff --git a/fastreid/modeling/backbones/resnext.py b/fastreid/modeling/backbones/resnext.py index 6d61125..be8e8b4 100644 --- a/fastreid/modeling/backbones/resnext.py +++ b/fastreid/modeling/backbones/resnext.py @@ -259,6 +259,7 @@ def init_pretrained_weights(key): cached_file = os.path.join(model_dir, filename) if not os.path.exists(cached_file): + logger.info(f"Pretrain model don't exist, downloading from {model_urls[key]}") if comm.is_main_process(): gdown.download(model_urls[key], cached_file, quiet=False)