test: add restart policy to keras native tests
@@ -78,7 +78,7 @@ class NativeImplementations:
"batches_per_step": 4,
"checkpoint_storage": experiment.shared_fs_checkpoint_config(),
"searcher": {"name": "single", "max_steps": 1, "metric": "val_accuracy"},
- "max_restarts": 0,
+ "max_restarts": 2,
},
num_expected_steps_per_trial=1,
num_expected_trials=1,
@@ -98,7 +98,7 @@ class NativeImplementations:
"checkpoint_storage": experiment.shared_fs_checkpoint_config(),
"searcher": {"name": "single", "max_steps": 1, "metric": "val_accuracy"},
"resources": {"slots_per_trial": 2},
- "max_restarts": 0,
+ "max_restarts": 2,
},
num_expected_steps_per_trial=1,
num_expected_trials=1,
@@ -113,7 +113,7 @@ class NativeImplementations:
"batches_per_step": 4,
"checkpoint_storage": experiment.shared_fs_checkpoint_config(),
"searcher": {"name": "single", "max_steps": 1, "metric": "val_accuracy"},
- "max_restarts": 0,
+ "max_restarts": 2,
},
num_expected_steps_per_trial=1,
num_expected_trials=1,
@@ -127,7 +127,7 @@ class NativeImplementations:
"batches_per_step": 4,
"checkpoint_storage": experiment.shared_fs_checkpoint_config(),
"searcher": {"name": "single", "max_steps": 1, "metric": "val_accuracy"},
- "max_restarts": 0,
+ "max_restarts": 2,
},
num_expected_steps_per_trial=1,
num_expected_trials=1,
