add , in init

2026-09-24 23:33:08 +08:00 · 2021-09-01 17:04:18 +08:00
parent 3374ad7620
commit e10a178010
1 changed files with 2 additions and 1 deletions
@@ -25,7 +25,8 @@ class PSELoss(nn.Layer):
                 ohem_ratio=3,
                 kernel_sample_mask='pred',
                 reduction='sum',
-                 eps=1e-6**kwargs):
+                 eps=1e-6,
+                 **kwargs):
        """Implement PSE Loss.
        """
        super(PSELoss, self).__init__()