Skip to content

GitLab

Commit 504f2550 authored by ghpark's avatar ghpark
Browse files

Rename test

parent 315506c9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/projects/detr/tasks/detection_test.py
View file @ 504f2550
...@@ -116,7 +116,7 @@ class DetectionTest(tf.test.TestCase): ...@@ -116,7 +116,7 @@ class DetectionTest(tf.test.TestCase):
state = task.aggregate_logs(step_outputs=logs) state = task.aggregate_logs(step_outputs=logs)
task.reduce_aggregated_logs(state) task.reduce_aggregated_logs(state)
class DetectionTest_tfds(tf.test.TestCase): class DetectionTFDSTest(tf.test.TestCase):
def test_train_step(self): def test_train_step(self):
config = detr_cfg.DetrTask( config = detr_cfg.DetrTask(
...@@ -177,7 +177,7 @@ class DetectionTest_tfds(tf.test.TestCase): ...@@ -177,7 +177,7 @@ class DetectionTest_tfds(tf.test.TestCase):
state = task.aggregate_logs(step_outputs=logs) state = task.aggregate_logs(step_outputs=logs)
task.reduce_aggregated_logs(state) task.reduce_aggregated_logs(state)
class DetectionTest_tfrecord(tf.test.TestCase): class DetectionTFRecordTest(tf.test.TestCase):
def test_train_step(self): def test_train_step(self):
config = detr_cfg.DetrTask( config = detr_cfg.DetrTask(
......
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