Yolov5-distillation-train-inference icon indicating copy to clipboard operation
Yolov5-distillation-train-inference copied to clipboard

Published 20 hours ago • verified publisher icon Sharpiless

这个v5是5.0么？

Open Bobby-youngking opened this issue 4 years ago • 5 comments

Bobby-youngking avatar May 13 '21 06:05 Bobby-youngking

是5.0

Sharpiless avatar May 14 '21 03:05 Sharpiless

我发现这个网络训练起来非常慢，博主实验过了吗

Bobby-youngking avatar May 14 '21 06:05 Bobby-youngking

正在实验并查找原因

Sharpiless avatar May 14 '21 06:05 Sharpiless

我查找到了两点，不知道对不对，第一点：teacher网络本身占用了太大的资源；第二点：训练batch的问题，学生必须等老师训练完一个batch才能开始训练，而老师必须等学生训练完当前batch后才能开始下一个batch

Bobby-youngking avatar May 14 '21 06:05 Bobby-youngking

这个问题暂时没法解决，因为增强后的图像必须重新计算其logits

Sharpiless avatar May 14 '21 07:05 Sharpiless