From 38fb46b147f2572495a3b2bf70c4e0f9be2d347a Mon Sep 17 00:00:00 2001 From: Andrzej Pomirski Date: Mon, 28 Sep 2020 00:53:28 +0200 Subject: [PATCH] Change comment to indicate the difference --- .../keras/graph_transformations/model_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/model_transformer.py b/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/model_transformer.py index 95dc8f972..5dff80668 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/model_transformer.py +++ b/tensorflow_model_optimization/python/core/quantization/keras/graph_transformations/model_transformer.py @@ -416,7 +416,7 @@ def _add_replacement_nodes(first_layer_removed_index, replacement_nodes): def _weight_name(name): """Extracts the weight name by removing layer from TF variable name. - For example, returns 'kernel:0' for 'dense_2/kernel:0'. + For example, returns 'conv1d_25/kernel:0' for 'quantize_annotate_2/conv1d_25/kernel:0'. Args: name: TensorFlow variable name.