fix typo in cosmargin

2021-07-13 07:54:29 +00:00 · 2021-07-13 07:54:29 +00:00 · c965193f36
parent a0941fb44a
commit c965193f36
1 changed files with 1 additions and 1 deletions
--- 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(