当前位置:   article > 正文

RT-DETR 训练时无法打印 GFLOPS【官方解决方案来了】_detr训练时打印

detr训练时打印

仓库问题地址:https://github.com/ultralytics/ultralytics/issues/6846

关于这个问题目前有了一个解决方案,LINTAO5835 这位同学反馈说 :

找到 ultralytics/utils/torch_utils.py 中的 get_flops 方法,将 stride 改成 640 就可以了,即将:

stride = max(int(model.stride.max()), 32
    声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/434359
    推荐阅读
    相关标签
      

    闽ICP备14008679号