diff --git a/manifests/v1alpha3/katib-controller/trialTemplateConfigmap.yaml b/manifests/v1alpha3/katib-controller/trialTemplateConfigmap.yaml index 67ec76c00cd..c0e9f0009d6 100644 --- a/manifests/v1alpha3/katib-controller/trialTemplateConfigmap.yaml +++ b/manifests/v1alpha3/katib-controller/trialTemplateConfigmap.yaml @@ -15,6 +15,14 @@ data: spec: containers: - name: {{.Trial}} - image: alpine + image: docker.io/katib/mxnet-mnist-example + command: + - "python" + - "/mxnet/example/image-classification/train_mnist.py" + - "--batch-size=64" + {{- with .HyperParameters}} + {{- range .}} + - "{{.Name}}={{.Value}}" + {{- end}} + {{- end}} restartPolicy: Never -