Few changes

kubeflow · Mar 8, 2021 · a2a58ff · a2a58ff
1 parent 4135314
commit a2a58ff
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/examples/v1beta1/pytorch-mnist/mnist.py b/examples/v1beta1/pytorch-mnist/mnist.py
@@ -111,7 +111,7 @@ def main():
                             default=dist.Backend.GLOO)
     args = parser.parse_args()
 
-    # Use this format (%Y-%m-%dT%H:%M:%SZ) to record timestamp of the metrics
+    # Use this format (%Y-%m-%dT%H:%M:%SZ) to record timestamp of the metrics.
     # If log_path is empty print log to StdOut, otherwise print log to the file.
     if args.log_path == "":
         logging.basicConfig(

diff --git a/operators/katib-controller/files/pytorchJobTemplate.yaml b/operators/katib-controller/files/pytorchJobTemplate.yaml
@@ -15,6 +15,7 @@ spec:
               command:
                 - "python3"
                 - "/opt/pytorch-mnist/mnist.py"
+                - "--epochs=1"
                 - "--lr=${trialParameters.learningRate}"
                 - "--momentum=${trialParameters.momentum}"
     Worker:

diff --git a/test/scripts/v1beta1/setup-katib.sh b/test/scripts/v1beta1/setup-katib.sh
@@ -79,9 +79,9 @@ sed -i -e "s@docker.io/kubeflowkatib\/trial-darts-cnn-cifar10:.*@${ECR_REGISTRY}
 
 # TODO (andreyvelich): For testing
 cat examples/v1beta1/random-example.yaml
-cat examples/v1beta1/early-stopping/median.stop.yaml
+cat examples/v1beta1/early-stopping/median-stop.yaml
 cat examples/v1beta1/grid-example.yaml
-cat examples/v1beta1/pytorch-example.yaml
+cat examples/v1beta1/pytorchjob-example.yaml
 cat examples/v1beta1/nas/darts-example-cpu.yaml
 cat examples/v1beta1/nas/enas-example-cpu.yaml