diff options
Diffstat (limited to 'experiments/crossover_artifact_validation_smoke.py')
| -rw-r--r-- | experiments/crossover_artifact_validation_smoke.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/experiments/crossover_artifact_validation_smoke.py b/experiments/crossover_artifact_validation_smoke.py index e267832..f410a3e 100644 --- a/experiments/crossover_artifact_validation_smoke.py +++ b/experiments/crossover_artifact_validation_smoke.py @@ -44,6 +44,15 @@ def main(): "allowed_physical_gpus": [5, 7], } validate(launch, "resnet_p1_launch", "abc") + transformer_launch = copy.deepcopy(launch) + transformer_launch["num_jobs"] = 27 + validate(transformer_launch, "transformer_p1_launch", "abc") + r2_launch = copy.deepcopy(transformer_launch) + r2_launch["stage"] = "r2" + validate(r2_launch, "resnet_r2_launch", "abc") + t2_launch = copy.deepcopy(transformer_launch) + t2_launch["stage"] = "t2" + validate(t2_launch, "transformer_t2_launch", "abc") bad_launch = copy.deepcopy(launch) bad_launch["allowed_physical_gpus"] = [0, 1] must_fail(bad_launch, "resnet_p1_launch", "abc") |