/usr/local/lib64/python3.6/site-packages/torch/utils/data/__pycache__
NameSizeModeActions
backward_compatibility.cpython-36.pyc4740644editdlrm
dataloader.cpython-36.pyc241450644editdlrm
dataloader_experimental.cpython-36.pyc42840644editdlrm
dataset.cpython-36.pyc170510644editdlrm
distributed.cpython-36.pyc48510644editdlrm
graph.cpython-36.pyc12660644editdlrm
sampler.cpython-36.pyc90120644editdlrm
sharding.cpython-36.pyc10110644editdlrm
_decorator.cpython-36.pyc60310644editdlrm
_typing.cpython-36.pyc98520644editdlrm
__init__.cpython-36.pyc14770644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/utils/data/__pycache__/sharding.cpython-36.pyc (1011B)
3 úEg7ã@sddlZdd„ZdS)éNcs~tjjjj|ƒ}‡fdd„‰ˆ|ƒ}d}xP|D]H}t|dƒr.|jƒr.t|dƒr.|dk rftdd|d|ƒ‚|j||ƒ|}q.WdS)NcsJtƒ}x>|jƒD]2\}}|j|ƒˆ|ƒ}x|D]}|j|ƒq0WqW|S)N)ÚsetÚitemsÚadd)ÚgraphÚresultsÚdatapipeZ sub_graphZ sub_itemsÚitem)Útraverse_graph©úE/usr/local/lib64/python3.6/site-packages/torch/utils/data/sharding.pyr s  z&apply_sharding..traverse_graphÚ is_shardableÚapply_shardingzVThis implementation of sharding can be only applied once per instance of DataPipeline.zAlready applied tozwhile trying to apply to) ÚtorchÚutilsÚdatarZtraverseÚhasattrr Ú RuntimeErrorr )rZnum_of_instancesZ instance_idrZ all_pipesZalready_applied_toÚpiper )r r r s      r )Ztorch.utils.data.graphrr r r r r Ús