mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add logger for pretrain model downloading

pull/504/head
liaoxingyu 2021-06-08 15:57:47 +08:00
parent 7ed6240e2c
commit 4f90197336
3 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
fastreid/modeling/backbones/osnet.py
View File

@ -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)

1
fastreid/modeling/backbones/resnet.py
View File

@ -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)

1
fastreid/modeling/backbones/resnext.py
View File

@ -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)