diff --git a/ppcls/losses/emlloss.py b/ppcls/losses/emlloss.py index 2ce934572ccd8b1452e44fb0aecb6e68a94df82b..410e34780a3a117da152b6731444026a97467565 100644 --- a/ppcls/losses/emlloss.py +++ b/ppcls/losses/emlloss.py @@ -86,15 +86,4 @@ class EmlLoss(paddle.nn.Layer): loss = self.surrogate_function_stable(self.beta, theta, bias, self.thresh) loss = paddle.mean(loss) return {"emlloss": loss} - -if __name__=="__main__": - - metric = EmlLoss() - - np.random.seed(1) - features = np.random.randn(40, 32) - features = paddle.to_tensor(features, dtype="float32") - print(features) - loss = metric(features) - print(loss) \ No newline at end of file