From c965193f3607204b396c316a1a8a7dc37782a9b4 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Tue, 13 Jul 2021 07:54:29 +0000 Subject: [PATCH] fix typo in cosmargin --- ppcls/arch/gears/cosmargin.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/arch/gears/cosmargin.py b/ppcls/arch/gears/cosmargin.py index 1d3ff83a..51db5508 100644 --- 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( -- GitLab