tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝,通道剪枝,知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

Error report when using apex to accelerate training #89

Open CNSTJiatongLee opened 4 years ago

CNSTJiatongLee commented 4 years ago

When I use apex to speed up the training, I will raise the error that the cudnn algorithm cannot be found. When apex is manually closed, it runs normally. Use apex to accelerate normal in the ultralytics version

zbyuan commented 4 years ago

Cudnn version may be wrong