/usr/local/lib64/python3.6/site-packages/torch/utils/data/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/utils/data/__pycache__/sharding.cpython-36.pyc (1011B)
3
úEg7 ã @ s d dl Zdd„ ZdS )é Nc s~ t jjjj| ƒ}‡ fdd„‰ ˆ |ƒ}d }xP|D ]H}t|dƒr.|jƒ r.t|dƒr.|d k rftdd|d|ƒ‚|j||ƒ |}q.W d S )Nc sJ t ƒ }x>| jƒ D ]2\}}|j|ƒ ˆ |ƒ}x|D ]}|j|ƒ q0W qW |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Údatar ZtraverseÚhasattrr ÚRuntimeErrorr
)r Znum_of_instancesZinstance_idr Z all_pipesZalready_applied_toÚpiper
)r r r
s
r
)Ztorch.utils.data.graphr r
r
r
r
r Ú s