Unverified 提交 178c1095 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Add non-zero `da` `check_anchor_order()` condition (#7066)

上级 9e75cbf4
......@@ -20,7 +20,7 @@ def check_anchor_order(m):
a = m.anchors.prod(-1).mean(-1).view(-1) # mean anchor area per output layer
da = a[-1] - a[0] # delta a
ds = m.stride[-1] - m.stride[0] # delta s
if da.sign() != ds.sign(): # same order
if da and (da.sign() != ds.sign()): # same order
LOGGER.info(f'{PREFIX}Reversing anchor order')
m.anchors[:] = m.anchors.flip(0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论