Add val_step unittest

open-mmlab · Sep 25, 2021 · 0ecffa1 · 0ecffa1
1 parent eb87ace
commit 0ecffa1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/test_models/test_segmentors/utils.py b/tests/test_models/test_segmentors/utils.py
@@ -101,6 +101,14 @@ def _segmentor_forward_train_test(segmentor):
         imgs, img_metas, gt_semantic_seg=gt_semantic_seg, return_loss=True)
     assert isinstance(losses, dict)
 
+    # Test val_step
+    with torch.no_grad():
+        segmentor.eval()
+        data_batch = dict(
+            img=imgs, img_metas=img_metas, gt_semantic_seg=gt_semantic_seg)
+        outputs = segmentor.val_step(data_batch)
+        assert isinstance(outputs, dict)
+
     # Test forward simple test
     with torch.no_grad():
         segmentor.eval()