We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a4ba973 commit 3c5146fCopy full SHA for 3c5146f
1 file changed
tests/training/test_train_swa.py
@@ -501,11 +501,7 @@ def test_training_config_validation():
501
with pytest.raises(ValueError, match='n_epoch must be positive'):
502
TrainingConfig(n_epoch=0)
503
504
- # Invalid: swa_start must be between 0 and n_epoch
505
- with pytest.raises(ValueError, match='swa_start must be between'):
506
- TrainingConfig(n_epoch=100, swa_start=100)
507
-
508
+ with pytest.raises(ValueError, match='swa_start must be non-negative'):
509
TrainingConfig(n_epoch=100, swa_start=-1)
510
511
# Invalid: swa_freq must be positive
0 commit comments