File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -4422,12 +4422,12 @@ def validate_full_validation_config(
44224422 ema_full_validation_enabled = bool (validating .get ("ema_full_validation" , False ))
44234423 if not full_validation_enabled :
44244424 return
4425- if float (validating .get ("full_val_start" , 0.0 )) == 1.0 :
4426- return
44274425 if ema_full_validation_enabled and not training_params .get ("enable_ema" , False ):
44284426 raise ValueError (
44294427 "validating.ema_full_validation requires `training.enable_ema=true`."
44304428 )
4429+ if float (validating .get ("full_val_start" , 0.0 )) == 1.0 :
4430+ return
44314431
44324432 metric = str (validating .get ("validation_metric" , "E:MAE" ))
44334433 if not is_valid_full_validation_metric (metric ):
Original file line number Diff line number Diff line change @@ -300,7 +300,7 @@ def test_full_validator_lmdb_full_validation_iterates_nloc_groups(self) -> None:
300300 },
301301 validation_data = validation_data ,
302302 model = _DummyModel (),
303- train_infos = {},
303+ state_store = {},
304304 num_steps = 10 ,
305305 rank = 0 ,
306306 zero_stage = 0 ,
You can’t perform that action at this time.
0 commit comments