diff --git a/model_compression_toolkit/core/pytorch/graph_substitutions/substitutions/functional_linear.py b/model_compression_toolkit/core/pytorch/graph_substitutions/substitutions/functional_linear.py index 4765d910e..41ee11e7d 100644 --- a/model_compression_toolkit/core/pytorch/graph_substitutions/substitutions/functional_linear.py +++ b/model_compression_toolkit/core/pytorch/graph_substitutions/substitutions/functional_linear.py @@ -47,7 +47,7 @@ def substitute(self, Graph after applying the substitution. """ - # Create new node of layer convolution + # Create new node of layer Linear if 1 not in func_node.weights: Logger.critical(f'Weight input missing for node {func_node.name}.') # pragma: no cover # Extract index of kernel and bias according to tensor_input_allocs if they were input as kwargs. If