search

Linzaer / Ultra-Light-Fast-Generic-Face-Detector-1MB

💎1MB lightweight face detection model (1MB轻量级人脸检测模型)
MIT License
7.14k stars 1.54k forks source link

如何优化GPU训练速度 #294

Open cuiyong127 opened 9 months ago

cuiyong127 commented 9 months ago

请问由于这个模型比较小，现在采用3090训练，GPU利用率很低，大部分时间GPU利用率为0，加大batch size 也没有改善，反而占用了更多的GPU显存。应该是cpu或者硬盘读写速度不够，请问有什么方式可以优化？