Rename test_scheduler_outputs_equivalence in model tests. (open-mmlab…

…#451)
aravind-h-v · Sep 13, 2022 · e335f05 · e335f05
1 parent f7cd6b8
commit e335f05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
@@ -179,7 +179,7 @@ def test_ema_training(self):
         loss.backward()
         ema_model.step(model)
 
-    def test_scheduler_outputs_equivalence(self):
+    def test_outputs_equivalence(self):
         def set_nan_tensor_to_zero(t):
             # Temporary fallback until `aten::_index_put_impl_` is implemented in mps
             # Track progress in https://github.com/pytorch/pytorch/issues/77764