search

open-mmlab / mmcv

OpenMMLab Computer Vision Foundation
https://mmcv.readthedocs.io/en/latest/
Apache License 2.0
5.84k stars 1.63k forks source link

Any plans to support DALI? #1690

Open liming-ai opened 2 years ago

liming-ai commented 2 years ago

For some tasks (e.g. contrastive learning), data augmentation is very important and data augmentation on CPU is often very slow, are there any plans to support the use of DALI in the future?

zhouzaida commented 2 years ago

Yes, we've discussed data processing acceleration before, but we think it is not a high-priority plan. However, if you have any plans, feel free to create PR.

liming-ai commented 2 years ago

Yes, we've discussed data processing acceleration before, but we think it is not a high-priority plan. However, if you have any plans, feel free to create PR.

Hi @zhouzaida ,

Maybe this repo could be implemented in mmcv with DALI?

zhouzaida commented 2 years ago

Thank you for the reference, we will look into it.