frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add verbose parameter to torch.hub.list (#120717)

Browse Source 
This PR adds ```verbose``` to ```torch.hub.list``` to let  users being able to disable extraneous outputs.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/120717
Approved by: https://github.com/ezyang
This commit is contained in:
cyy
2024-03-01 07:39:44 +00:00
committed by PyTorch MergeBot
parent 63b259492a
commit 77ef9d4022
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
torch/hub.py
View File

@ -379,7 +379,7 @@ def set_dir(d):
_hub_dir = os.path.expanduser(d)
def list(github, force_reload=False, skip_validation=False, trust_repo=None):
def list(github, force_reload=False, skip_validation=False, trust_repo=None, verbose=True):
r"""
List all callable entrypoints available in the repo specified by ``github``.
@ -411,6 +411,9 @@ def list(github, force_reload=False, skip_validation=False, trust_repo=None):
v2.0.
Default is ``None`` and will eventually change to ``"check"`` in v2.0.
verbose (bool, optional): If ``False``, mute messages about hitting
local caches. Note that the message about first download cannot be
muted. Default is ``True``.
Returns:
list: The available callables entrypoint
@ -419,7 +422,7 @@ def list(github, force_reload=False, skip_validation=False, trust_repo=None):
>>> # xdoctest: +REQUIRES(env:TORCH_DOCTEST_HUB)
>>> entrypoints = torch.hub.list('pytorch/vision', force_reload=True)
"""
repo_dir = _get_cache_or_reload(github, force_reload, trust_repo, "list", verbose=True,
repo_dir = _get_cache_or_reload(github, force_reload, trust_repo, "list", verbose=verbose,
skip_validation=skip_validation)
with _add_to_sys_path(repo_dir):