快捷方式

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)])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源