From af7ec73439e75d4560f0c46072f5d9e743a2d624 Mon Sep 17 00:00:00 2001 From: Shashank Mittal Date: Tue, 3 Sep 2024 02:08:39 +0530 Subject: [PATCH] added e2e test to workflow Signed-off-by: Shashank Mittal --- .github/workflows/e2e-test-pytorch-mnist.yaml | 3 ++- examples/v1beta1/hp-tuning/hyperopt-distribution.yaml | 5 ++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/.github/workflows/e2e-test-pytorch-mnist.yaml b/.github/workflows/e2e-test-pytorch-mnist.yaml index 38059456a3b..3a5ff7e1bc0 100644 --- a/.github/workflows/e2e-test-pytorch-mnist.yaml +++ b/.github/workflows/e2e-test-pytorch-mnist.yaml @@ -40,6 +40,7 @@ jobs: # suggestion-hyperopt - "long-running-resume,from-volume-resume,median-stop" # others + - "grid,bayesian-optimization,tpe,multivariate-tpe,cma-es,hyperband" - "hyperopt-distribution" - "file-metrics-collector,pytorchjob-mnist" - - "median-stop-with-json-format,file-metrics-collector-with-json-format" + - "median-stop-with-json-format,file-metrics-collector-with-json-format" \ No newline at end of file diff --git a/examples/v1beta1/hp-tuning/hyperopt-distribution.yaml b/examples/v1beta1/hp-tuning/hyperopt-distribution.yaml index b845140542c..9b641174991 100644 --- a/examples/v1beta1/hp-tuning/hyperopt-distribution.yaml +++ b/examples/v1beta1/hp-tuning/hyperopt-distribution.yaml @@ -20,11 +20,14 @@ spec: feasibleSpace: min: "0.01" max: "0.05" + step: "0.01" + distribution: "uniform" - name: momentum parameterType: double feasibleSpace: min: "0.5" max: "0.9" + distribution: "logUniform" trialTemplate: primaryContainerName: training-container trialParameters: @@ -54,4 +57,4 @@ spec: limits: memory: "1Gi" cpu: "0.5" - restartPolicy: Never + restartPolicy: Never \ No newline at end of file