[Fix] Fix resnet audio ut (#2637)

open-mmlab · Sep 6, 2023 · dd1901a · dd1901a
1 parent 2ddf4b5
commit dd1901a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/models/backbones/test_resnet_audio.py b/tests/models/backbones/test_resnet_audio.py
@@ -3,13 +3,15 @@
 
 from mmaction.models import ResNetAudio
 from mmaction.testing import generate_backbone_demo_inputs
+from mmaction.utils import register_all_modules
 
 
 def test_resnet_audio_backbone():
     """Test ResNetAudio backbone."""
     input_shape = (1, 1, 16, 16)
     spec = generate_backbone_demo_inputs(input_shape)
     # inference
+    register_all_modules()
     audioonly = ResNetAudio(50, None)
     audioonly.init_weights()
     audioonly.train()