Suggestion: Use Nvidia DALI as the dataloader

cavalleria / cavaface

face recognition training project(pytorch)

MIT License

456 stars 88 forks source link

Open xsacha opened 3 years ago

xsacha commented 3 years ago

Would improve the input speed.

cavalleria commented 3 years ago

Thank for your suggestion. did you compare acc and speed when use NVIDIA DALI?

xsacha commented 3 years ago

No, I haven't sorry.

anxiangsir commented 3 years ago

dali must convert insightface rec to a new rec.

John1231983 commented 3 years ago

FYI, DaLI is slower than pytorch in distributed training. I tried it