No public description

PiperOrigin-RevId: 639121310
pyoung2778 · May 31, 2024 · e3254a6 · e3254a6
1 parent c9d1883
commit e3254a6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/official/common/flags.py b/official/common/flags.py
@@ -112,3 +112,12 @@ def define_flags():
 
   flags.DEFINE_string(
       'tpu_platform', default=None, help='TPU platform type.')
+
+  flags.DEFINE_string(
+      'tfhub_handle',
+      None,
+      'TFHub handle for publishing the model to TFHub. The model '
+      'is published to TFHub if this flag is set. Please use a '
+      'TFHubPusher (go/tflex/standard_components/pusher) component if '
+      'running in TFleX.',
+  )