search

WenmuZhou / PytorchOCR

基于Pytorch的OCR工具库，支持常用的文字检测和识别算法
1.39k stars 305 forks source link

MobileNetV3 small CRNN训练跑不满GPU #256

Closed huangxin168 closed 1 year ago

huangxin168 commented 1 year ago

大佬，MobileNetV3 small CRNN训练batchsize合workers各种设置，固态硬盘，16G单卡跑不满，占用率50~90之间变化，请问是因为模型太小导致的吗？

huangxin168 commented 1 year ago

更新： 换resnet后gpu占用率80~99之间变化，好很多