Add parameter detection test for lightweight

chaiNNer-org · Nov 20, 2023 · 1380d04 · 1380d04
1 parent ae483c1
commit 1380d04
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_KBNet.py b/tests/test_KBNet.py
@@ -23,4 +23,5 @@ def test_KBCNN_load():
         lambda: KBNet_s(width=32, middle_blk_num=3),
         lambda: KBNet_s(width=32, enc_blk_nums=[1, 2, 3, 4], dec_blk_nums=[1, 2, 3, 4]),
         lambda: KBNet_s(width=32, ffn_scale=3),
+        lambda: KBNet_s(width=32, lightweight=True),
     )