RT-DETR 训练时无法打印 GFLOPS【官方解决方案来了】
2024-01-08 04:46:56
仓库问题地址:https://github.com/ultralytics/ultralytics/issues/6846
关于这个问题目前有了一个解决方案,LINTAO5835 这位同学反馈说 :
找到 ultralytics/utils/torch_utils.py
中的 get_flops
方法,将 stride
改成 640
就可以了,即将:
stride = max(int(model.stride.max()), 32
文章来源:https://blog.csdn.net/weixin_43694096/article/details/135394091
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!