周波涛 1 년 전
부모
커밋
f62b46fd7f
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      paddlers/models/ppdet/modeling/assigners/uniform_assigner.py

+ 1 - 1
paddlers/models/ppdet/modeling/assigners/uniform_assigner.py

@@ -30,7 +30,7 @@ def batch_p_dist(x, y, p=2):
     """
     x = x.unsqueeze(1)
     diff = x - y
-    return paddle.norm(diff, p=p, axis=list(range(2, diff.dim())))
+    return paddle.linalg.vector_norm(diff, p=p, axis=list(range(2, diff.dim())))
 
 
 @register