Skip to content

GitLab

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

Temporarily disables the train_test. 

PiperOrigin-RevId: 370834123
parent c2722e76
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/projects/assemblenet/train_test.py
View file @ ff2ca513
...@@ -48,7 +48,7 @@ class TrainTest(tf.test.TestCase): ...@@ -48,7 +48,7 @@ class TrainTest(tf.test.TestCase):
# pylint: enable=g-complex-comprehension # pylint: enable=g-complex-comprehension
tfexample_utils.dump_to_tfrecord(self._data_path, tf_examples=examples) tfexample_utils.dump_to_tfrecord(self._data_path, tf_examples=examples)
def test_run(self): def disable_test_run(self):
saved_flag_values = flagsaver.save_flag_values() saved_flag_values = flagsaver.save_flag_values()
train_lib.tfm_flags.define_flags() train_lib.tfm_flags.define_flags()
FLAGS.mode = 'train' FLAGS.mode = 'train'
......
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