Fix paddle.no_grad in moe example. (#6715)

guoshengCS · web-flow · commit 09329bd614ed · 2023-08-14T17:09:39.000+08:00
diff --git a/examples/language_model/moe/dygraph/framework/group_sharded.py b/examples/language_model/moe/dygraph/framework/group_sharded.py
@@ -43,7 +43,7 @@
 
 
 class ClipGradForShardedMOEByGlobalNorm(ClipGradForMOEByGlobalNorm):
-    @paddle.no_grad
+    @paddle.no_grad()
     def _dygraph_clip(self, params_grads):
         normal_params_grads = []
         moe_params_grads = []
