[Fix] fix ut for bmn and bsn (#1966)

open-mmlab · Oct 14, 2022 · da65bdf · da65bdf
1 parent d675cad
commit da65bdf
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/tests/models/localizers/test_bmn.py b/tests/models/localizers/test_bmn.py
@@ -33,7 +33,7 @@ def get_localization_data_sample():
 @pytest.mark.skipif(platform.system() == 'Windows', reason='Windows mem limit')
 def test_bmn_loss():
     model_cfg = get_localizer_cfg(
-        'bmn/bmn_400x100_2x8_9e_activitynet_feature.py')
+        'bmn/bmn_2xb8-400x100-9e_activitynet-feature.py')
 
     if 0 and torch.cuda.is_available():
         raw_feature = [torch.rand(400, 100).cuda()]
@@ -53,7 +53,7 @@ def test_bmn_loss():
 @pytest.mark.skipif(platform.system() == 'Windows', reason='Windows mem limit')
 def test_bmn_predict():
     model_cfg = get_localizer_cfg(
-        'bmn/bmn_400x100_2x8_9e_activitynet_feature.py')
+        'bmn/bmn_2xb8-400x100-9e_activitynet-feature.py')
 
     if 0 and torch.cuda.is_available():
         localizer_bmn = MODELS.build(model_cfg.model).cuda()
@@ -73,7 +73,7 @@ def test_bmn_predict():
 @pytest.mark.skipif(platform.system() == 'Windows', reason='Windows mem limit')
 def test_bmn_tensor():
     model_cfg = get_localizer_cfg(
-        'bmn/bmn_400x100_2x8_9e_activitynet_feature.py')
+        'bmn/bmn_2xb8-400x100-9e_activitynet-feature.py')
 
     if 0 and torch.cuda.is_available():
         localizer_bmn = MODELS.build(model_cfg.model).cuda()

diff --git a/tests/models/localizers/test_pem.py b/tests/models/localizers/test_pem.py
@@ -28,7 +28,7 @@ def get_localization_data_sample():
 @pytest.mark.skipif(platform.system() == 'Windows', reason='Windows mem limit')
 def test_pem():
     model_cfg = get_localizer_cfg(
-        'bsn/bsn_pem_400x100_1x16_20e_activitynet_feature.py')
+        'bsn/bsn_pem_1xb16-400x100-20e_activitynet-feature.py')
 
     localizer_pem = MODELS.build(model_cfg.model)
     raw_features = [torch.rand(100, 32)] * 8

diff --git a/tests/models/localizers/test_tem.py b/tests/models/localizers/test_tem.py
@@ -33,7 +33,7 @@ def get_localization_data_sample():
 @pytest.mark.skipif(platform.system() == 'Windows', reason='Windows mem limit')
 def test_tem():
     model_cfg = get_localizer_cfg(
-        'bsn/bsn_tem_400x100_1x16_20e_activitynet_feature.py')
+        'bsn/bsn_tem_1xb16-400x100-20e_activitynet-feature.py')
 
     localizer_tem = MODELS.build(model_cfg.model)
     raw_feature = torch.rand(8, 400, 100)