fix comments

sony · Apr 17, 2024 · 87613c2 · 87613c2
1 parent 0019a9f
commit 87613c2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/model_compression_toolkit/core/common/substitutions/remove_identity.py b/model_compression_toolkit/core/common/substitutions/remove_identity.py
@@ -5,7 +5,7 @@
 def remove_identity_node(graph: Graph,
                          node: BaseNode) -> Graph:
     """
-    The method to perform the substitution of the `torch.nn.Identity` node by
+    The method to perform the substitution of the identity node by
     reconnecting its input directly to its output, effectively removing the node
     from the graph.
 

diff --git a/model_compression_toolkit/core/keras/graph_substitutions/substitutions/remove_identity.py b/model_compression_toolkit/core/keras/graph_substitutions/substitutions/remove_identity.py
@@ -25,7 +25,7 @@
 
 class RemoveIdentity(common.BaseSubstitution):
     """
-    Remove `torch.nn.Identity` layers from the graph.
+    Remove Identity layers from the graph.
     """
 
     def __init__(self):
@@ -36,7 +36,7 @@ def substitute(self,
                    graph: Graph,
                    node: BaseNode) -> Graph:
         """
-        The method to perform the substitution of the `torch.nn.Identity` node by
+        The method to perform the substitution of the identity keras node by
         reconnecting its input directly to its output, effectively removing the node
         from the graph.