do not misconstrue Optimizer slot values as embeddings (#3679)

tensorflow · Jun 1, 2020 · 4c50549 · 4c50549
1 parent 12f6234
commit 4c50549
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensorboard/plugins/projector/projector_plugin.py b/tensorboard/plugins/projector/projector_plugin.py
@@ -387,6 +387,10 @@ def _augment_configs_with_checkpoint_info(self):
             for tensor_name, tensor_shape in var_map.items():
                 if len(tensor_shape) != 2:
                     continue
+                # Optimizer slot values are the same shape as embeddings
+                # but are not embeddings.
+                if ".OPTIMIZER_SLOT" in tensor_name:
+                    continue
                 embedding = self._get_embedding(tensor_name, config)
                 if not embedding:
                     embedding = config.embeddings.add()