Rename test_scheduler_outputs_equivalence in model tests. (#451)
This commit is contained in:
parent
f7cd6b87e1
commit
e335f05fb1
|
@ -179,7 +179,7 @@ class ModelTesterMixin:
|
|||
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
|
||||
|
|
Loading…
Reference in New Issue