Internal change

PiperOrigin-RevId: 373393602

official/vision/beta/configs/image_classification_test.py

@@ -39,6 +39,7 @@ class ImageClassificationConfigTest(tf.test.TestCase, parameterized.TestCase):
     self.assertIsInstance(config.task.model,
                           exp_cfg.ImageClassificationModel)
     self.assertIsInstance(config.task.train_data, exp_cfg.DataConfig)
+    config.validate()
     config.task.train_data.is_training = None
     with self.assertRaises(KeyError):
       config.validate()

official/vision/beta/configs/maskrcnn_test.py

@@ -29,6 +29,7 @@ class MaskRCNNConfigTest(tf.test.TestCase, parameterized.TestCase):
       ('fasterrcnn_resnetfpn_coco',),
       ('maskrcnn_resnetfpn_coco',),
       ('maskrcnn_spinenet_coco',),
+      ('cascadercnn_resnetfpn_coco',),
   )
   def test_maskrcnn_configs(self, config_name):
     config = exp_factory.get_exp_config(config_name)
@@ -36,6 +37,7 @@ class MaskRCNNConfigTest(tf.test.TestCase, parameterized.TestCase):
     self.assertIsInstance(config.task, exp_cfg.MaskRCNNTask)
     self.assertIsInstance(config.task.model, exp_cfg.MaskRCNN)
     self.assertIsInstance(config.task.train_data, exp_cfg.DataConfig)
+    config.validate()
     config.task.train_data.is_training = None
     with self.assertRaisesRegex(KeyError, 'Found inconsistncy between key'):
       config.validate()

official/vision/beta/configs/retinanet_test.py

@@ -23,19 +23,20 @@ from official.vision import beta
 from official.vision.beta.configs import retinanet as exp_cfg
 
 
-class MaskRCNNConfigTest(tf.test.TestCase, parameterized.TestCase):
+class RetinaNetConfigTest(tf.test.TestCase, parameterized.TestCase):
 
   @parameterized.parameters(
       ('retinanet_resnetfpn_coco',),
       ('retinanet_spinenet_coco',),
       ('retinanet_spinenet_mobile_coco',),
   )
-  def test_maskrcnn_configs(self, config_name):
+  def test_retinanet_configs(self, config_name):
     config = exp_factory.get_exp_config(config_name)
     self.assertIsInstance(config, cfg.ExperimentConfig)
     self.assertIsInstance(config.task, exp_cfg.RetinaNetTask)
     self.assertIsInstance(config.task.model, exp_cfg.RetinaNet)
     self.assertIsInstance(config.task.train_data, exp_cfg.DataConfig)
+    config.validate()
     config.task.train_data.is_training = None
     with self.assertRaisesRegex(KeyError, 'Found inconsistncy between key'):
       config.validate()

official/vision/beta/configs/semantic_segmentation_test.py

@@ -36,6 +36,7 @@ class ImageSegmentationConfigTest(tf.test.TestCase, parameterized.TestCase):
     self.assertIsInstance(config.task.model,
                           exp_cfg.SemanticSegmentationModel)
     self.assertIsInstance(config.task.train_data, exp_cfg.DataConfig)
+    config.validate()
     config.task.train_data.is_training = None
     with self.assertRaises(KeyError):
       config.validate()

official/vision/beta/configs/video_classification_test.py

@@ -35,6 +35,7 @@ class VideoClassificationConfigTest(tf.test.TestCase, parameterized.TestCase):
     self.assertIsInstance(config.task, exp_cfg.VideoClassificationTask)
     self.assertIsInstance(config.task.model, exp_cfg.VideoClassificationModel)
     self.assertIsInstance(config.task.train_data, exp_cfg.DataConfig)
+    config.validate()
     config.task.train_data.is_training = None
     with self.assertRaises(KeyError):
       config.validate()