CacheTimeout¶
- class torchdata.dataloader2.adapter.CacheTimeout(timeout=None)¶
CacheTimeout DataPipes 适配器允许控制图中所有现有 EndOnDiskCacheHolder(
end_caching
)的超时。当缓存管道执行时间过长(例如,文件下载速度慢)时很有用。- 参数:
timeout – int - 并行进程等待缓存文件出现的秒数。
示例
dp = IterableWrapper(range(size)).shuffle() dl = DataLoader2(dp, [CacheTimeout(600)])