Merge branch 'dev-3.x' of github.com:open-mmlab/mmdetection into ssod…

…_tutorial
open-mmlab · Aug 27, 2022 · 63ca912 · 63ca912
2 parents 7f0c66b + e405153
commit 63ca912
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/test_engine/test_hooks/test_mean_teacher_hook.py b/tests/test_engine/test_hooks/test_mean_teacher_hook.py
@@ -26,6 +26,7 @@ def __init__(self):
 
     def forward(self, inputs, data_samples, mode='tensor'):
         labels = torch.stack(data_samples)
+        inputs = torch.stack(inputs)
         outputs = self.linear(inputs)
         if mode == 'tensor':
             return outputs

diff --git a/tests/test_engine/test_runner/test_loops.py b/tests/test_engine/test_runner/test_loops.py
@@ -25,6 +25,7 @@ def __init__(self):
 
     def forward(self, inputs, data_samples, mode='tensor'):
         labels = torch.stack(data_samples)
+        inputs = torch.stack(inputs)
         outputs = self.linear(inputs)
         if mode == 'tensor':
             return outputs
@@ -71,7 +72,7 @@ def __len__(self):
         return self.data.size(0)
 
     def __getitem__(self, index):
-        return dict(inputs=self.data[index], data_sample=self.label[index])
+        return dict(inputs=self.data[index], data_samples=self.label[index])
 
 
 class TestTeacherStudentValLoop(TestCase):
@@ -82,7 +83,7 @@ def setUp(self):
     def tearDown(self):
         self.temp_dir.cleanup()
 
-    def test_mean_teacher_hook(self):
+    def test_teacher_student_val_loop(self):
         device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
         model = ToyModel2().to(device)
         evaluator = Mock()