https://github.com/kubeflow/katib
Raw File
Tip revision: a8709d09191eb39cbf08a71e0700eaf58e8918e2 authored by DeeperMind on 23 January 2019, 02:58:53 UTC
Add NAS RL Suggestion
Tip revision: a8709d0
gpu-example.yaml
apiVersion: "kubeflow.org/v1alpha1"
kind: StudyJob
metadata:
  namespace: kubeflow
  labels:
    controller-tools.k8s.io: "1.0"
  name: gpu-example
spec:
  studyName: gpu-example
  owner: crd
  optimizationtype: maximize
  objectivevaluename: Validation-accuracy
  optimizationgoal: 0.99
  metricsnames:
    - accuracy
  parameterconfigs:
    - name: --lr
      parametertype: double
      feasible:
        min: "0.01"
        max: "0.03"
    - name: --num-layers
      parametertype: int
      feasible:
        min: "2"
        max: "3"
    - name: --optimizer
      parametertype: categorical
      feasible:
        list:
        - sgd
        - adam
        - ftrl
  workerSpec:
    goTemplate:
        templatePath: "/worker-template/gpuWorkerTemplate.yaml"
  suggestionSpec:
    suggestionAlgorithm: "random"
    requestNumber: 3
back to top