Skip to content

GitLab

Commit 8843bb24 authored by Jiageng Zhang's avatar Jiageng Zhang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 463990853
parent 1d8d8ede
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/beta/projects/panoptic_maskrcnn/modeling/panoptic_deeplab_model_test.py
View file @ 8843bb24
...@@ -19,6 +19,7 @@ import numpy as np ...@@ -19,6 +19,7 @@ import numpy as np
import tensorflow as tf import tensorflow as tf
from tensorflow.python.distribute import combinations from tensorflow.python.distribute import combinations
from tensorflow.python.distribute import test_util
from official.vision.beta.projects.panoptic_maskrcnn.modeling import panoptic_deeplab_model from official.vision.beta.projects.panoptic_maskrcnn.modeling import panoptic_deeplab_model
from official.vision.beta.projects.panoptic_maskrcnn.modeling.heads import panoptic_deeplab_heads from official.vision.beta.projects.panoptic_maskrcnn.modeling.heads import panoptic_deeplab_heads
from official.vision.beta.projects.panoptic_maskrcnn.modeling.layers import panoptic_deeplab_merge from official.vision.beta.projects.panoptic_maskrcnn.modeling.layers import panoptic_deeplab_merge
...@@ -182,4 +183,4 @@ class PanopticDeeplabNetworkTest(parameterized.TestCase, tf.test.TestCase): ...@@ -182,4 +183,4 @@ class PanopticDeeplabNetworkTest(parameterized.TestCase, tf.test.TestCase):
if __name__ == '__main__': if __name__ == '__main__':
tf.test.main() test_util.main()
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