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

单机多卡稀疏训练 #80

Closed lwbmowgli closed 4 years ago

lwbmowgli commented 4 years ago

image 我的机器上四块2080ti,当进行稀疏训练的时候报错信息如上,但是我指定一个GPU运行则没问题,您的工程目前还不支持多卡稀疏训练吗?