Commit 764fb2e
committed
test: cap neuron gated model training to 1 epoch in slow test
test_gated_model_training_v2_neuron ran a full Llama-2-7B neuron
training on ml.trn1.32xlarge with no epoch cap, taking ~170+ min and
timing out the 3h slow-tests build. Cap training to a single epoch to
exercise only the train/deploy/predict flow, matching the existing
epochs="1" / max_steps="1" pattern used by the other JumpStart
estimator slow tests.
X-AI-Prompt: Add epochs=1 hyperparameter to test_gated_model_training_v2_neuron to fix slow-tests timeout
X-AI-Tool: kiro-cli1 parent d0cbf41 commit 764fb2e
1 file changed
Lines changed: 3 additions & 0 deletions
Lines changed: 3 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
215 | 215 | | |
216 | 216 | | |
217 | 217 | | |
| 218 | + | |
| 219 | + | |
| 220 | + | |
218 | 221 | | |
219 | 222 | | |
220 | 223 | | |
| |||
0 commit comments