pytorch dali 加速

shishi_m037192554發表於2020-12-21

使用dali加速,前提是 gpu沒有跑滿,不然效果也不大
+apex 混合精度訓練

1、【Pytorch】nvidia-dali——一種加速資料增強的方法
https://blog.csdn.net/weixin_42028608/article/details/105564060

2、官方自定義資料介面使用介紹
https://github.com/NVIDIA/DALI/blob/master/docs/examples/frameworks/pytorch/pytorch-external_input.ipynb

在這裡插入圖片描述

相關文章