From 65031531ee2d40e81931b7a6d5b1c696ee09261d Mon Sep 17 00:00:00 2001
From: cir7 <33249023+cir7@users.noreply.github.com>
Date: Fri, 30 Sep 2022 16:46:30 +0800
Subject: [PATCH] [Fix] fix ut for bmn and bsn (#1966)

---
 tests/models/localizers/test_bmn.py | 6 +++---
 tests/models/localizers/test_pem.py | 2 +-
 tests/models/localizers/test_tem.py | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/tests/models/localizers/test_bmn.py b/tests/models/localizers/test_bmn.py
index 884c13c5c8..f39a96e81a 100644
--- 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
index b2b6292f4b..623e4acff7 100644
--- 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
index 54c4b88d87..a7e9d9c236 100644
--- 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)