We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6423024 commit 1e90d97Copy full SHA for 1e90d97
tests/ignite/engine/test_create_supervised.py
@@ -457,7 +457,7 @@ def test_create_supervised_trainer_amp_error(mock_torch_cuda_amp_module):
457
_test_create_supervised_trainer_wrong_accumulation(trainer_device="cpu", amp_mode="amp")
458
with pytest.raises(ImportError, match="Please install torch>=1.12.0 to use amp_mode='amp'."):
459
_test_create_supervised_trainer(amp_mode="amp")
460
- with pytest.raises(ImportError, match="Please install torch>=1.12.0 to use amp_mode='amp'."):
+ with pytest.raises(ImportError, match="Please install torch>=1.6.0 to use scaler argument."):
461
_test_create_supervised_trainer(amp_mode="amp", scaler=True)
462
463
0 commit comments