U <ºcƒã@s8dZGdd„deƒZGdd„deƒZGdd„deƒZdS)zÈ"Contains definitions of the methods used by the _BaseDataLoaderIter to fetch data from an iterable-style or map-style dataset. This logic is shared in both single- and multi-processing data loading. c@seZdZdd„Zdd„ZdS)Ú_BaseDatasetFetchercCs||_||_||_||_dS©N)ÚdatasetÚauto_collationÚ collate_fnÚ drop_last©Úselfrrrr©r úA/tmp/pip-unpacked-wheel-gikjz4vx/torch/utils/data/_utils/fetch.pyÚ__init__sz_BaseDatasetFetcher.__init__cCs tƒ‚dSr)ÚNotImplementedError)rÚpossibly_batched_indexr r r Úfetchsz_BaseDatasetFetcher.fetchN)Ú__name__Ú __module__Ú __qualname__r rr r r r rsrcs$eZdZ‡fdd„Zdd„Z‡ZS)Ú_IterableDatasetFetchercs*tt|ƒ ||||¡t|ƒ|_d|_dS)NF)Úsuperrr ÚiterÚ dataset_iterÚendedr©Ú __class__r r r s ÿ z _IterableDatasetFetcher.__init__c Cs|jr t‚|jr|g}|D]:}z| t|jƒ¡Wqtk rPd|_YqTYqXqt|ƒdksv|jr†t|ƒt|ƒkr†t‚n t|jƒ}| |¡S)NTé) rÚ StopIterationrÚappendÚnextrÚlenrr)rr ÚdataÚ_r r r rs$ ÿÿ z_IterableDatasetFetcher.fetch©rrrr rÚ __classcell__r r rr rs rcs$eZdZ‡fdd„Zdd„Z‡ZS)Ú_MapDatasetFetchercstt|ƒ ||||¡dSr)rr"r rrr r r 0s  ÿz_MapDatasetFetcher.__init__csPˆjr:sz,_MapDatasetFetcher.fetch..)rÚhasattrrr#r)rr rr r&r r5s  z_MapDatasetFetcher.fetchr r r rr r"/s r"N)Ú__doc__Úobjectrrr"r r r r Ús