add_module_in_init_

PaddlePaddle · Feb 16, 2022 · 0890aa9 · 0890aa9
1 parent eb3c7d0
commit 0890aa9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/paddle/nn/__init__.py b/python/paddle/nn/__init__.py
@@ -307,4 +307,5 @@ def weight_norm(*args):
            'MaxUnPool2D',
            'MaxUnPool3D',
            'HingeEmbeddingLoss',
+           'Identity',
 ]