"src/lib/vscode:/vscode.git/clone" did not exist on "33b9629164437f5e72d268aa0134114580b87ba5"
Commit 71277b49 authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 446509422
parent 96a8d744
......@@ -82,6 +82,7 @@ class TrainTest(parameterized.TestCase, tf.test.TestCase):
})
FLAGS.params_override = params_override
with train_lib.train.gin.unlock_config():
train_lib.train.main('unused_args')
FLAGS.mode = 'eval'
......
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