add ut

open-mmlab · May 23, 2023 · 973e9d7 · 973e9d7
1 parent 4eff271
commit 973e9d7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_models/test_models.py b/tests/test_models/test_models.py
@@ -34,6 +34,7 @@ class Cfg:
         backbone=mmpretrain.models.VisionTransformer,
         forward=False,
         backward=False),
+    Cfg(name='hivit-tiny-p16_16xb64_in1k', backbone=mmpretrain.models.HiViT),
 ]