diff --git a/RELEASE.md b/RELEASE.md index a5959bef9..409230eff 100644 --- a/RELEASE.md +++ b/RELEASE.md @@ -30,6 +30,14 @@ Keras pruning API: * Tested against TensorFlow 1.14.0, 2.0.0, and nightly, and Python 3. +# TensorFlow Model Optimization next release TBD + +Keras clustering API: + +* Added *ClusteringSummaries* to create additional output for the clustering +progress for TensorBoard. +* Tested against TensorFlow 1.14.0, 2.0.0, and nightly, and Python 3. + # TensorFlow Model Optimization 0.5.0 TFMOT 0.5.0 adds some additional features for Quantization Aware Training. QAT diff --git a/tensorflow_model_optimization/python/core/api/clustering/keras/__init__.py b/tensorflow_model_optimization/python/core/api/clustering/keras/__init__.py index 8e28f9ff5..86b6dc0f8 100644 --- a/tensorflow_model_optimization/python/core/api/clustering/keras/__init__.py +++ b/tensorflow_model_optimization/python/core/api/clustering/keras/__init__.py @@ -21,4 +21,5 @@ from tensorflow_model_optimization.python.core.clustering.keras.cluster import strip_clustering from tensorflow_model_optimization.python.core.clustering.keras.cluster_config import CentroidInitialization +from tensorflow_model_optimization.python.core.clustering.keras.clustering_callbacks import ClusteringSummaries # pylint: enable=g-bad-import-order diff --git a/tensorflow_model_optimization/python/core/clustering/keras/BUILD b/tensorflow_model_optimization/python/core/clustering/keras/BUILD index 9669aff73..835940e74 100644 --- a/tensorflow_model_optimization/python/core/clustering/keras/BUILD +++ b/tensorflow_model_optimization/python/core/clustering/keras/BUILD @@ -12,6 +12,7 @@ py_library( srcs_version = "PY3", deps = [ ":cluster", + ":clustering_callbacks", "//tensorflow_model_optimization/python/core/clustering/keras/experimental", ], )