本文简要介绍python语言中 torchrec.datasets.utils.ParallelReadConcat
的用法。
用法:
class torchrec.datasets.utils.ParallelReadConcat(*datapipes: torch.utils.data.dataset.IterDataPipe, dp_selector: typing.Callable[[typing.Sequence[torch.utils.data.dataset.IterDataPipe]], typing.Sequence[torch.utils.data.dataset.IterDataPipe]] = <function _default_dp_selector>)
基础:
torch.utils.data.dataset.IterDataPipe
ParallelReadConcat
。可迭代 DataPipe 连接多个可迭代 DataPipe。与 DataLoader 一起使用时,将数据管道的子集分配给每个 DataLoader 工作线程以允许并行读取。 :param datapipes: IterDataPipe 要读取的实例。 :param dp_selector:每个DataLoader工作人员将使用该函数来确定要读取的数据管道的子集:param:
例子:
datapipes = [ criteo_terabyte( (f"/home/local/datasets/criteo/shard_{idx}.tsv",), ) .batch(100) .collate() for idx in range(4) ] dataloader = DataLoader( ParallelReadConcat(*datapipes), num_workers=4, batch_size=None )
相关用法
- Python PyTorch ParagraphAggregator用法及代码示例
- Python PyTorch ParameterList用法及代码示例
- Python PyTorch ParameterDict用法及代码示例
- Python PyTorch ParquetDataFrameLoader用法及代码示例
- Python PyTorch Pareto用法及代码示例
- Python PyTorch PackageImporter.id用法及代码示例
- Python PyTorch PackageExporter.register_extern_hook用法及代码示例
- Python PyTorch PairwiseDistance用法及代码示例
- Python PyTorch PackageExporter.register_intern_hook用法及代码示例
- Python PyTorch PackageExporter.close用法及代码示例
- Python PyTorch PackageExporter.register_mock_hook用法及代码示例
- Python PyTorch Proxy用法及代码示例
- Python PyTorch PoissonNLLLoss用法及代码示例
- Python PyTorch PixelShuffle用法及代码示例
- Python PyTorch Pipe用法及代码示例
- Python PyTorch PooledEmbeddingsAllToAll用法及代码示例
- Python PyTorch Poisson用法及代码示例
- Python PyTorch Perceptron用法及代码示例
- Python PyTorch PixelUnshuffle用法及代码示例
- Python PyTorch PitchShift用法及代码示例
- Python PyTorch PooledEmbeddingsReduceScatter用法及代码示例
- Python PyTorch PReLU用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torchrec.datasets.utils.ParallelReadConcat。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。