/usr/local/lib64/python3.6/site-packages/torch/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/__pycache__/hub.cpython-36.pyc (19171B)
3
EgK[ @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl
mZmZ d dlmZ yd dlmZ W nJ ek
r yd dlmZ W n$ ek
r G dd deZY nX Y nX ejdZd Zd
Zd Zd Zd
Z dZ dZ da dd Z dd Z!dd Z"dd Z#dd Z$dd Z%d d Z&d d Z'dBd"d#Z(d$d% Z)d&d' Z*d(d) Z+d*d+ Z,d,d- Z-dCd.d/Z.dDd0d1Z/d2d!d d!d3d4d5Z0d6d7 Z1dEd8d9Z2dFd:d;Z3d
d? Z5dGd@dAZ6dS )H N) HTTPError)urlopenRequest)urlparse)tqdmc @ s6 e Zd Zd
ddZdd Zdd Zd d
Zd d ZdS )r NFc C s || _ || _d| _d S )Nr )totaldisablen)selfr r unit
unit_scale unit_divisor r 5/usr/local/lib64/python3.6/site-packages/torch/hub.py__init__ s z
tqdm.__init__c C sh | j r
d S | j|7 _| jd kr8tjjdj| j n"tjjdjd| j t| j tjj d S )Nz
{0:.1f} bytesz
{0:.1f}%d ) r r r sysstderrwriteformatfloatflush)r
r r r r update! s
"z tqdm.updatec C s
d| _ d S )NT)r )r
r r r close, s z
tqdm.closec C s | S )Nr )r
r r r __enter__/ s ztqdm.__enter__c C s | j r
d S tjjd d S )N
)r r r r )r
exc_typeexc_valexc_tbr r r __exit__2 s z
tqdm.__exit__)NFNNN)__name__
__module__ __qualname__r r r r r r r r r r s
r z-([a-f0-9]*)\.Z GITHUB_TOKENZ
TORCH_HOMEXDG_CACHE_HOMEz~/.cache dependenciesz
hubconf.pyi c C sN dd l }ddlm} |jj| |}|jj|}t|j|s>t|jj | |S )Nr )Loader)
importlib.utilZ
importlib.abcr% utilspec_from_file_locationmodule_from_spec
isinstanceloaderAssertionError exec_module)namepath importlibr% specmoduler r r
import_moduleF s r3 c C s2 t jj| r.t jj| r$t j| n
tj| d S )N)osr/ existsisfileremoveshutilrmtree)r/ r r r _remove_if_existsP s r: c C s dj | ||S )Nz'https://github.com/{}/{}/archive/{}.zip)r )
repo_owner repo_namebranchr r r _git_archive_linkX s r> c C s |t | krd S t| |S )N)dirgetattr)r2 func_namer r r _load_attr_from_module\ s rB c C s* t jjt jtt jjt jttd} | S )Ntorch)r4 r/
expandusergetenvENV_TORCH_HOMEjoinENV_XDG_CACHE_HOMEDEFAULT_CACHE_DIR)Z
torch_homer r r _get_torch_homec s
rJ c C s d| kr| j d\}}n
| d }}|j d\}}|d kry*td| d| d
d}W d Q R X W n4 tk
r } z|jdkrd}n W Y d d }~X nX |||fS )N:/zhttps://github.com/z /tree/main/maini Zmaster)splitr r code)githubZ repo_infor= r; r< er r r _parse_repo_infok s
rR c
C s* t | }|j j|jjdS Q R X d S )Nzutf-8)r readdecodeheadersget_content_charset)urlrr r r _read_url s
rY c
C s ddi}t jjt}|d k r*d| |d< xd| d| dd| d| dfD ]p}d }xf|d
7 }| d | }tjtt||d }|sP x.|D ]&} | d
|ks| d d j|rd S qW qZW qPW t d| d| d| dd S )NAcceptz application/vnd.github.v3+jsonztoken
Authorizationz https://api.github.com/repos/rL z /branchesz/tagsr z?per_page=100&page=)rU r. commitshaz Cannot find z in https://github.com/zX. If it's a commit from a forked repo, please call hub.load() with forked repo directly.)
r4 environgetENV_GITHUB_TOKENjsonloadsrY r
startswith
ValueError)
r; r< r= rU tokenZ
url_prefixpagerW responsebrr r r _validate_not_a_forked_repo s$
rj TFc
C s: t }tjj|s tj| t| \}}}|jdd}tjj|dj|||g} | o`tjj| }
|
r~|r|tj j
dj | n|st ||| tjj||d } t
| t|||} tj j
dj | | t| | dd tj| 4}
|
j d j}tjj||}t
| |
j| W d Q R X t
| t
| tj|| | S ) NrL _zUsing cache found in {}
z.zipzDownloading: "{}" to {}
F)progressr )get_dirr4 r/ r5 makedirsrR replacerG r r r r rj r: r> download_url_to_filezipfileZipFileinfolistfilename
extractallr8 move)rP force_reloadverboseskip_validationhub_dirr; r< r= Z
normalized_brrepo_dirZ use_cache cached_filerW Zcached_zipfileZextraced_repo_nameZextracted_repor r r _get_cache_or_reload s4
r} c C s dd l }|jj| d k S )Nr )r& r' find_spec)r. r0 r r r _check_module_exists s r c C s@ t | t}|d k r s z'_check_dependencies..zMissing dependencies: {}z, )rB VAR_DEPENDENCYlen RuntimeErrorr rG )mr$ missing_depsr r r _check_dependencies s
r c C sH t |tstdt| t| |}|d ks6t| rDtdj||S )Nz8Invalid input: model should be a string of function namez"Cannot find callable {} in hubconf)r* strre r rB callabler r )r modelfuncr r r _load_entry_from_hubconf s
r c C s0 t jdrtjd tdk r tS t jjt dS )a
Get the Torch Hub cache directory used for storing downloaded models & weights.
If :func:`~torch.hub.set_dir` is not called, default path is ``$TORCH_HOME/hub`` where
environment variable ``$TORCH_HOME`` defaults to ``$XDG_CACHE_HOME/torch``.
``$XDG_CACHE_HOME`` follows the X Design Group specification of the Linux
filesystem layout, with a default value ``~/.cache`` if the environment
variable is not set.
Z TORCH_HUBz:TORCH_HUB is deprecated, please use env TORCH_HOME insteadNZhub)r4 rE warningswarn_hub_dirr/ rG rJ r r r r rm s
rm c C s | a dS )z
Optionally set the Torch Hub directory used to save downloaded models & weights.
Args:
d (string): path to a local folder to save downloaded models & weights.
N)r )dr r r set_dir s r c s\ t | |d|d}tjjd| tjj|t}tt| tjj| fddt D }|S )a"
List all callable entrypoints available in the repo specified by ``github``.
Args:
github (string): a string with format "repo_owner/repo_name[:tag_name]" with an optional
tag/branch. If ``tag_name`` is not specified, the default branch is assumed to be ``main`` if
it exists, and otherwise ``master``.
Example: 'pytorch/vision:0.10'
force_reload (bool, optional): whether to discard the existing cache and force a fresh download.
Default is ``False``.
skip_validation (bool, optional): if ``False``, torchhub will check that the branch or commit
specified by the ``github`` argument properly belongs to the repo owner. This will make
requests to the GitHub API; you can specify a non-default GitHub token by setting the
``GITHUB_TOKEN`` environment variable. Default is ``False``.
Returns:
list: The available callables entrypoint
Example:
>>> entrypoints = torch.hub.list('pytorch/vision', force_reload=True)
T)rx ry r c s* g | ]"}t t |r|jd r|qS )rk )r r@ rd )r f)
hub_moduler r r - s zlist..)
r} r r/ insertr4 rG MODULE_HUBCONFr3 r7 r? )rP rw ry r{ hubconf_pathZ entrypointsr )r r list s
r c C sR t | |d|d}tjjd| tjj|t}tt|}tjj| t ||}|j
S )a
Show the docstring of entrypoint ``model``.
Args:
github (string): a string with format with an optional
tag/branch. If ``tag_name`` is not specified, the default branch is assumed to be ``main`` if
it exists, and otherwise ``master``.
Example: 'pytorch/vision:0.10'
model (string): a string of entrypoint name defined in repo's ``hubconf.py``
force_reload (bool, optional): whether to discard the existing cache and force a fresh download.
Default is ``False``.
skip_validation (bool, optional): if ``False``, torchhub will check that the branch or commit
specified by the ``github`` argument properly belongs to the repo owner. This will make
requests to the GitHub API; you can specify a non-default GitHub token by setting the
``GITHUB_TOKEN`` environment variable. Default is ``False``.
Example:
>>> print(torch.hub.help('pytorch/vision', 'resnet18', force_reload=True))
T)rx ry r ) r} r r/ r r4 rG r r3 r7 r __doc__)rP r rw ry r{ r r entryr r r help2 s
r rP )sourcerw rx ry c O sL |j }|dkr td| d|dkr6t| |||} t| |f||}|S )a(
Load a model from a github repo or a local directory.
Note: Loading a model is the typical use case, but this can also be used to
for loading other objects such as tokenizers, loss functions, etc.
If ``source`` is 'github', ``repo_or_dir`` is expected to be
of the form ``repo_owner/repo_name[:tag_name]`` with an optional
tag/branch.
If ``source`` is 'local', ``repo_or_dir`` is expected to be a
path to a local directory.
Args:
repo_or_dir (string): If ``source`` is 'github',
this should correspond to a github repo with format ``repo_owner/repo_name[:tag_name]`` with
an optional tag/branch, for example 'pytorch/vision:0.10'. If ``tag_name`` is not specified,
the default branch is assumed to be ``main`` if it exists, and otherwise ``master``.
If ``source`` is 'local' then it should be a path to a local directory.
model (string): the name of a callable (entrypoint) defined in the
repo/dir's ``hubconf.py``.
*args (optional): the corresponding args for callable ``model``.
source (string, optional): 'github' or 'local'. Specifies how
``repo_or_dir`` is to be interpreted. Default is 'github'.
force_reload (bool, optional): whether to force a fresh download of
the github repo unconditionally. Does not have any effect if
``source = 'local'``. Default is ``False``.
verbose (bool, optional): If ``False``, mute messages about hitting
local caches. Note that the message about first download cannot be
muted. Does not have any effect if ``source = 'local'``.
Default is ``True``.
skip_validation (bool, optional): if ``False``, torchhub will check that the branch or commit
specified by the ``github`` argument properly belongs to the repo owner. This will make
requests to the GitHub API; you can specify a non-default GitHub token by setting the
``GITHUB_TOKEN`` environment variable. Default is ``False``.
**kwargs (optional): the corresponding kwargs for callable ``model``.
Returns:
The output of the ``model`` callable when called with the given
``*args`` and ``**kwargs``.
Example:
>>> # from a github repo
>>> repo = 'pytorch/vision'
>>> model = torch.hub.load(repo, 'resnet50', pretrained=True)
>>> # from a local directory
>>> path = '/some/local/path/pytorch/vision'
>>> model = torch.hub.load(path, 'resnet50', pretrained=True)
rP localzUnknown source: "z&". Allowed values: "github" | "local".)rP r )lowerre r} _load_local)Z repo_or_dirr r rw rx ry argskwargsr r r loadS s 3r c O sJ t jjd| tjj| t}tt|}t||}|||}t jj| |S )a
Load a model from a local directory with a ``hubconf.py``.
Args:
hubconf_dir (string): path to a local directory that contains a
``hubconf.py``.
model (string): name of an entrypoint defined in the directory's
``hubconf.py``.
*args (optional): the corresponding args for callable ``model``.
**kwargs (optional): the corresponding kwargs for callable ``model``.
Returns:
a single model with corresponding pretrained weights.
Example:
>>> path = '/some/local/path/pytorch/vision'
>>> model = _load_local(path, 'resnet50', pretrained=True)
r ) r r/ r r4 rG r r3 r r7 )Z hubconf_dirr r r r r r r r r r s
r c C s d}t | ddid}t|}|j }t|dr:|jd}n
|jd}|dk rdt|dkrdt|d }tj j
|}tj j |} t j
d| d }
z|dk rtj } t|| d
d d d
N} xF|jd}
t|
dkrP |
j|
|dk r| j|
| jt|
qW W dQ R X |
j |dk rD| j }|dt| |krDtdj||tj|
j| W d|
j tj j|
jrztj |
j X dS )a] Download object at the given URL to a local path.
Args:
url (string): URL of the object to download
dst (string): Full path where object will be saved, e.g. ``/tmp/temporary_file``
hash_prefix (string, optional): If not None, the SHA256 downloaded file should start with ``hash_prefix``.
Default: None
progress (bool, optional): whether or not to display a progress bar to stderr
Default: True
Example:
>>> torch.hub.download_url_to_file('https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth', '/tmp/temporary_file')
Nz
User-Agentz torch.hub)rU
getheaderszContent-Lengthr F)deleter? BTi )r r r r r
i z,invalid hash value (expected "{}", got "{}")) r r infohasattrr get_allr intr4 r/ rD dirnametempfileNamedTemporaryFilehashlibsha256r rS r r r hexdigestr r r8 rv r. r5 r7 )rW dst hash_prefixrl file_sizerequmetacontent_lengthZdst_dirr r Zpbarbufferdigestr r r rp sF
rp c C s t jd t| ||| d S )Nztorch.hub._download_url_to_file has been renamed to torch.hub.download_url_to_file to be a public API, _download_url_to_file will be removed in after 1.3 release)r r rp )rW r r rl r r r _download_url_to_file s r c C s6 t j| r2t j| j }t|dko0|d j S dS )Nr\ r F)rq
is_zipfilerr rs r is_dir)rt rs r r r _is_legacy_zip_format s
r c C sl t jd tj| D}|j }t|dkr2td|j| |d j}t j
j ||}W d Q R X t j
||dS )NzFalling back to the old format < 1.6. This support will be deprecated in favor of default zipfile format introduced in 1.6. Please redo torch.save() to save it in the new zipfile format.r\ z0Only one file(not dir) is allowed in the zipfiler ) map_location)r r rq rr rs r r ru rt r4 r/ rG rC r )rt model_dirr r membersZ
extraced_nameZextracted_filer r r _legacy_zip_load s
r c
C s t jdrtjd |dkr0t }t jj|d}yt j| W n2 tk
rp } z|j t j
kr^n W Y dd}~X nX t | }t jj |j} |dk r|} t jj|| }
t jj
|
stjjdj| |
d} |rtj| } | r| jdnd} t| |
| |d t|
r t|
||S tj|
|dS ) a_ Loads the Torch serialized object at the given URL.
If downloaded file is a zip file, it will be automatically
decompressed.
If the object is already present in `model_dir`, it's deserialized and
returned.
The default value of ``model_dir`` is ``/checkpoints`` where
``hub_dir`` is the directory returned by :func:`~torch.hub.get_dir`.
Args:
url (string): URL of the object to download
model_dir (string, optional): directory in which to save the object
map_location (optional): a function or a dict specifying how to remap storage locations (see torch.load)
progress (bool, optional): whether or not to display a progress bar to stderr.
Default: True
check_hash(bool, optional): If True, the filename part of the URL should follow the naming convention
``filename-.ext`` where ```` is the first eight or more
digits of the SHA256 hash of the contents of the file. The hash is used to
ensure unique names and to verify the contents of the file.
Default: False
file_name (string, optional): name for the downloaded file. Filename from ``url`` will be used if not set.
Example:
>>> state_dict = torch.hub.load_state_dict_from_url('https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth')
ZTORCH_MODEL_ZOOz@TORCH_MODEL_ZOO is deprecated, please use env TORCH_HOME insteadNZ checkpointszDownloading: "{}" to {}
r\ )rl )r )r4 rE r r rm r/ rG rn OSErrorerrnoEEXISTr basenamer5 r r r r
HASH_REGEXsearchgrouprp r r rC r )
rW r r rl Z
check_hash file_namerz rQ partsrt r| r rX r r r load_state_dict_from_url s4
r )TF)FF)FF)NT)NT)NNTFN)7r r rb r4 rer8 r r rC r rq urllib.errorr urllib.requestr r urllib.parser Z tqdm.autor ImportErrorobjectcompiler ra rF rH rI r r ZREAD_DATA_CHUNKr r3 r: r> rB rJ rR rY rj r} r r r rm r r r r r rp r r r r r r r r sh !
1
$
!@
<