Internal change

PiperOrigin-RevId: 366883391

official/common/flags.py

@@ -25,10 +25,14 @@ def define_flags():
   flags.DEFINE_enum(
       'mode',
       default=None,
-      enum_values=['train', 'eval', 'train_and_eval',
-                   'continuous_eval', 'continuous_train_and_eval'],
+      enum_values=[
+          'train', 'eval', 'train_and_eval', 'continuous_eval',
+          'continuous_train_and_eval', 'train_and_validate'
+      ],
       help='Mode to run: `train`, `eval`, `train_and_eval`, '
-      '`continuous_eval`, and `continuous_train_and_eval`.')
+      '`continuous_eval`, `continuous_train_and_eval` and '
+      '`train_and_validate` (which is not implemented in '
+      'the open source version).')
 
   flags.DEFINE_string(
       'model_dir',
@@ -78,7 +82,8 @@ def define_flags():
     pass
 
   flags.DEFINE_string(
-      'tpu', default=None,
+      'tpu',
+      default=None,
       help='The Cloud TPU to use for training. This should be either the name '
       'used when creating the Cloud TPU, or a grpc://ip.address.of.tpu:8470 '
       'url.')