Merge pull request #8604 from Ruomei:toupstream/clusteringexample

PiperOrigin-RevId: 329801211

official/requirements.txt

@@ -9,7 +9,7 @@ psutil>=5.4.3
 py-cpuinfo>=3.3.0
 scipy>=0.19.1
 tensorflow-hub>=0.6.0
-tensorflow-model-optimization>=0.2.1
+tensorflow-model-optimization>=0.4.1
 tensorflow-datasets
 tensorflow-addons
 dataclasses

official/vision/image_classification/resnet/common.py

@@ -353,6 +353,13 @@ def define_pruning_flags():
   flags.DEFINE_integer('pruning_frequency', 100, 'Frequency for pruning.')
 
 
+def define_clustering_flags():
+  """Define flags for clustering methods."""
+  flags.DEFINE_string('clustering_method', None,
+                      'None (no clustering) or selective_clustering '
+                      '(cluster last three Conv2D layers of the model).')
+
+
 def get_synth_input_fn(height,
                        width,
                        num_channels,