Skip to content

GitLab

Commit 4c514dc6 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 366883391
parent 6f0b84d0
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 4 deletions
official/common/flags.py
View file @ 4c514dc6
...@@ -25,10 +25,14 @@ def define_flags(): ...@@ -25,10 +25,14 @@ def define_flags():
flags.DEFINE_enum( flags.DEFINE_enum(
'mode', 'mode',
default=None, default=None,
enum_values=['train', 'eval', 'train_and_eval', enum_values=[
'continuous_eval', 'continuous_train_and_eval'], 'train', 'eval', 'train_and_eval', 'continuous_eval',
'continuous_train_and_eval', 'train_and_validate'
],
help='Mode to run: `train`, `eval`, `train_and_eval`, ' 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( flags.DEFINE_string(
'model_dir', 'model_dir',
...@@ -78,7 +82,8 @@ def define_flags(): ...@@ -78,7 +82,8 @@ def define_flags():
pass pass
flags.DEFINE_string( flags.DEFINE_string(
'tpu', default=None, 'tpu',
default=None,
help='The Cloud TPU to use for training. This should be either the name ' 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 ' 'used when creating the Cloud TPU, or a grpc://ip.address.of.tpu:8470 '
'url.') 'url.')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment