[Fix]: Change UT of cae

open-mmlab · May 5, 2022 · 59792f5 · 59792f5
1 parent 4fba323
commit 59792f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_models/test_algorithms/test_cae.py b/tests/test_models/test_algorithms/test_cae.py
@@ -7,7 +7,8 @@
 from mmselfsup.models.algorithms import CAE
 
 # model settings
-backbone = dict(type='CAEViT', arch='b', patch_size=16, init_values=0.1)
+backbone = dict(
+    type='CAEViT', arch='b', patch_size=16, init_values=0.1, qkv_bias=False)
 neck = dict(
     type='CAENeck',
     patch_size=16,