Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 446509422
parent a9860bfd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/projects/yt8m/train_test.py
View file @ 7147e1ca
...@@ -82,7 +82,8 @@ class TrainTest(parameterized.TestCase, tf.test.TestCase): ...@@ -82,7 +82,8 @@ class TrainTest(parameterized.TestCase, tf.test.TestCase):
}) })
FLAGS.params_override = params_override FLAGS.params_override = params_override
train_lib.train.main('unused_args') with train_lib.train.gin.unlock_config():
train_lib.train.main('unused_args')
FLAGS.mode = 'eval' 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