- 博客(2)
- 收藏
- 关注
RSS订阅原创 YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information，最新发布，超越yolov8
官方是提供了4个不同规模的模型，S、M、C、E,感觉接下来又有新的paper可以发了(不是。从github发布的指标图看，也超过了前一段时间华为发布的goldyolo。不过新发布的bug估计会很多，还是先跑跑看看。
2024-02-22 09:55:39
539
原创 对剪枝完的yolov8进行finetune时遇到RuntimeError: Expected all tensors to be on the same device的问题
该错误表明张量位于不同设备上（一个在 CUDA 上，另一个在 CPU 上）。这可能由于两个张量不在同一设备上引起的。通常，PyTorch要求操作的张量应在相同的设备上，否则会出现类似的错误。self.proj 可能不在与 pred_dist 相同的设备上。这可能是因为 self.proj 被指定在 CPU 上，而 pred_dist 在 GPU 上（或反之）。具体剪枝步骤参考https://blog.csdn.net/qq_40672115/article/details/130175558。
2023-11-09 16:14:45
651
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人