fix typo in cosmargin

c965193f · cuicheng01 · a0941fb4 · c965193f
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppcls/arch/gears/cosmargin.py ppcls/arch/gears/cosmargin.py +1 -1

未找到文件。
--- a/ppcls/arch/gears/cosmargin.py
+++ b/ppcls/arch/gears/cosmargin.py
@@ -38,7 +38,7 @@ class CosMargin(paddle.nn.Layer):

        input_norm = paddle.sqrt(
            paddle.sum(paddle.square(input), axis=1, keepdim=True))
-        input = paddle.divide(input, x_norm)
+        input = paddle.divide(input, input_norm)

        weight = self.fc.weight
        weight_norm = paddle.sqrt(