此版本训练太慢，如何解决？

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

Open zhuhuangru opened 4 years ago

zhuhuangru commented 4 years ago

原版Darknet 一般设置为batchsize 64,sub 64，训练起来相对比较快。这个版本由于内存原因，batchsize只能设置为6，训练非常慢，请问有什么解决办法吗？

tanluren commented 4 years ago

sub64比6快？看看能不能装apex