为什么我在使用 RT-DETR 训练时无法打印 GFLOPS，那么如何获取模型的 GFLOPS？

发布时间：2024年01月04日

仓库问题地址：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
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若内容造成侵权/违法违规/事实不符，请联系我的编程经验分享网邮箱：chenni525@qq.com进行投诉反馈，一经查实，立即删除！