forked from mojokb/kubeflow-book
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfashion-katib-random-v2.yaml
73 lines (73 loc) · 1.88 KB
/
fashion-katib-random-v2.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
apiVersion: "kubeflow.org/v1alpha3"
kind: Experiment
metadata:
namespace: dudaji # 사용하는 namespace 명으로 바꾸어야합니다.
labels:
controller-tools.k8s.io: "1.0"
name: fashion-mnist-cpu-experiment-v2-1
spec:
parallelTrialCount: 5
maxTrialCount: 50
maxFailedTrialCount: 3
objective:
type: maximize
goal: 0.99
objectiveMetricName: Validation-accuracy
additionalMetricNames:
- accuracy
- loss
- Validation-loss
algorithm:
algorithmName: random
trialTemplate:
goTemplate:
rawTemplate: |-
apiVersion: batch/v1
kind: Job
metadata:
name: {{.Trial}}
namespace: {{.NameSpace}}
spec:
template:
spec:
containers:
- name: {{.Trial}}
image: kubeflow-registry.default.svc.cluster.local:30000/fairing-job:506FE0D9
command:
- "python"
- "/app/fashion-mnist-katib.py"
{{- with .HyperParameters}}
{{- range .}}
- "{{.Name}}={{.Value}}"
{{- end}}
{{- end}}
restartPolicy: Never
parameters:
- name: --learning_rate
parameterType: double
feasibleSpace:
min: "0.0005"
max: "0.0015"
- name: --dropout_rate
parameterType: double
feasibleSpace:
min: "0.1"
max: "0.9"
- name: --layer
parameterType: int
feasibleSpace:
min: "1"
max: "5"
- name: --epoch
parameterType: int
feasibleSpace:
min: "5"
max: "15"
- name: --act
parameterType: categorical
feasibleSpace:
list: # relu, sigmoid, softmax, tanh
- "relu"
- "sigmoid"
- "softmax"
- "tanh"