Skip to content

GitLab

Commit 3046cda7 authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 440953577
parent 22bd174f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/projects/qat/vision/tasks/retinanet_test.py
View file @ 3046cda7
...@@ -60,6 +60,7 @@ class RetinaNetTaskTest(parameterized.TestCase, tf.test.TestCase): ...@@ -60,6 +60,7 @@ class RetinaNetTaskTest(parameterized.TestCase, tf.test.TestCase):
config.task.validation_data.shuffle_buffer_size = 2 config.task.validation_data.shuffle_buffer_size = 2
config.task.validation_data.input_path = test_tfrecord_file config.task.validation_data.input_path = test_tfrecord_file
config.task.train_data.input_path = test_tfrecord_file config.task.train_data.input_path = test_tfrecord_file
config.task.annotation_file = None
config.train_steps = 1 config.train_steps = 1
task = retinanet.RetinaNetTask(config.task) task = retinanet.RetinaNetTask(config.task)
......
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