【笔记】pytorch加载数据 时 如何优化加速：NVIDIA DALI是一个GPU加速的数据增强和图像加载库，支持单个和批处理图像的解码、缩放、Crop、颜色空间转换等

 

 

 

 

 

pipeline = DataPipeline(**dataset_dict)pipeline.build()loader = DALIGenericIterator(pipeline, ["imgs", "labels"], img_len, fill_last_batch=False)
for j in range(epoch):  for i, data in enumerate(loader):    x = data[0]["imgs"]    y = data[0]["labels"].cuda()    # train  loader.reset()