/usr/local/lib64/python3.6/site-packages/torch/nn/modules/__pycache__
NameSizeModeActions
activation.cpython-36.pyc476350644editdlrm
adaptive.cpython-36.pyc97250644editdlrm
batchnorm.cpython-36.pyc310390644editdlrm
channelshuffle.cpython-36.pyc19010644editdlrm
container.cpython-36.pyc279700644editdlrm
conv.cpython-36.pyc577890644editdlrm
distance.cpython-36.pyc36840644editdlrm
dropout.cpython-36.pyc103100644editdlrm
flatten.cpython-36.pyc58130644editdlrm
fold.cpython-36.pyc128070644editdlrm
instancenorm.cpython-36.pyc188080644editdlrm
lazy.cpython-36.pyc116900644editdlrm
linear.cpython-36.pyc101880644editdlrm
loss.cpython-36.pyc914650644editdlrm
module.cpython-36.pyc665300644editdlrm
normalization.cpython-36.pyc113600644editdlrm
padding.cpython-36.pyc222170644editdlrm
pixelshuffle.cpython-36.pyc43980644editdlrm
pooling.cpython-36.pyc532030644editdlrm
rnn.cpython-36.pyc446530644editdlrm
sparse.cpython-36.pyc209560644editdlrm
transformer.cpython-36.pyc207370644editdlrm
upsampling.cpython-36.pyc107800644editdlrm
utils.cpython-36.pyc25280644editdlrm
_functions.cpython-36.pyc54410644editdlrm
__init__.cpython-36.pyc52360644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/nn/modules/__pycache__/utils.cpython-36.pyc (2528B)
3 ûEgËã@s”ddlZddlmZddlmZmZmZdd„ZedƒZedƒZ edƒZ ed ƒZ d d „Z ee ee ee d œd d„Zeeefeddœdd„ZdS)éN)Úrepeat)ÚListÚDictÚAnycs‡fdd„}|S)Ncs$t|tjjƒrt|ƒStt|ˆƒƒS)N)Ú isinstanceÚ collectionsÚabcÚIterableÚtupler)Úx)Ún©úB/usr/local/lib64/python3.6/site-packages/torch/nn/modules/utils.pyÚparsesz_ntuple..parser )r rr )r rÚ_ntuples réééécst‡fdd„t|ƒDƒƒS)zµReverse the order of `t` and repeat each element for `n` times. This can be used to translate padding arg used by Conv and Pooling modules to the ones used by `F.pad`. c3s |]}tˆƒD] }|VqqdS)N)Úrange)Ú.0r Ú_)r r rú sz(_reverse_repeat_tuple..)r Úreversed)Útr r )r rÚ_reverse_repeat_tuplesr)Úout_sizeÚdefaultsÚreturncCsVt|tƒr|St|ƒt|ƒkr4tdjt|ƒdƒƒ‚dd„t||t|ƒ d…ƒDƒS)Nz%Input dimension should be at least {}rcSs g|]\}}|dk r|n|‘qS)Nr )rÚvÚdr r rú &sz&_list_with_default..)rÚintÚlenÚ ValueErrorÚformatÚzip)rrr r rÚ_list_with_defaults  r')Ú state_dictÚprefixrcCs˜t|jƒƒ}x4|D],}|j|ƒr|t|ƒd…}|j|ƒ||<qWd|kr”|d}x@t|jƒƒD]0}t|ƒdkrrq`|t|ƒd…}|j|ƒ||<q`WdS)a—Strip the prefix in state_dict in place, if any. ..note:: Given a `state_dict` from a DP/DDP model, a local model can load it by applying `consume_prefix_in_state_dict_if_present(state_dict, "module.")` before calling :meth:`torch.nn.Module.load_state_dict`. Args: state_dict (OrderedDict): a state-dict to be loaded to the model. prefix (str): prefix. NÚ _metadatar)ÚsortedÚkeysÚ startswithr#ÚpopÚlist)r(r)r,ÚkeyZnewkeyÚmetadatar r rÚ'consume_prefix_in_state_dict_if_present*s    r2)rÚ itertoolsrÚtypingrrrrZ_singleZ_pairZ_tripleZ _quadruplerr"r'Ústrr2r r r rÚs