Skip to content

GitLab

Commit f4e3d87b authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal changes. 

PiperOrigin-RevId: 314768990
parent 6a526b83
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 7 deletions
official/vision/detection/main.py
View file @ f4e3d87b
...@@ -42,15 +42,10 @@ from official.utils.misc import keras_utils ...@@ -42,15 +42,10 @@ from official.utils.misc import keras_utils
hyperparams_flags.initialize_common_flags() hyperparams_flags.initialize_common_flags()
flags_core.define_log_steps() flags_core.define_log_steps()
flags.DEFINE_bool( flags.DEFINE_bool('enable_xla', default=False, help='Enable XLA for GPU')
'enable_xla',
default=False,
help='Enable XLA for GPU')
flags.DEFINE_string( flags.DEFINE_string(
'mode', 'mode', default='train', help='Mode to run: `train` or `eval`.')
default='train',
help='Mode to run: `train`, `eval` or `train_and_eval`.')
flags.DEFINE_string( flags.DEFINE_string(
'model', default='retinanet', 'model', default='retinanet',
......
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