kumarx commited on
Commit
2f3e6a2
·
verified ·
1 Parent(s): b1e5abc

Test safe checkpoint cadence

Browse files
Files changed (1) hide show
  1. tests/test_training_modes.py +2 -0
tests/test_training_modes.py CHANGED
@@ -12,6 +12,8 @@ def test_training_modes_are_isolated_and_conservative():
12
  assert MODE_DEFAULTS["full"]["subset_size"] is None
13
  assert MODE_DEFAULTS["pilot"]["batch_size_per_gpu"] <= 1200
14
  assert MODE_DEFAULTS["full"]["grad_accumulation_steps"] >= 2
 
 
15
 
16
 
17
  def test_explicit_trainer_script_is_used(tmp_path: Path):
 
12
  assert MODE_DEFAULTS["full"]["subset_size"] is None
13
  assert MODE_DEFAULTS["pilot"]["batch_size_per_gpu"] <= 1200
14
  assert MODE_DEFAULTS["full"]["grad_accumulation_steps"] >= 2
15
+ assert MODE_DEFAULTS["smoke"]["save_every"] >= 100000
16
+ assert MODE_DEFAULTS["overfit"]["save_every"] >= 100000
17
 
18
 
19
  def test_explicit_trainer_script_is_used(tmp_path: Path):