-
Notifications
You must be signed in to change notification settings - Fork 0
/
polyaxon_hyperparams.yaml
43 lines (43 loc) · 1.42 KB
/
polyaxon_hyperparams.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
# See full specification at:
# https://docs.polyaxon.com/references/polyaxonfile-yaml-specification/
---
version: 1
kind: group
framework: keras
tags: ['mnist_keras']
hptuning:
concurrency: 3
matrix:
epochs:
values: [12, 50, 100]
log_learning_rate:
values: [-2, -3, -4]
optimizer:
values: ['adam', 'sgd']
loss_metric:
values: ['categorical_crossentropy']
first_layer_output:
values: [32, 60, 128]
kernel_size:
values: [2, 3, 4]
pool_size:
values: [2, 3]
layers:
values:
- 'conv2d:64,maxpooling2d,dropout:0.2,flatten,dense:128,dropout:0.5'
- 'conv2d:88,maxpooling2d,dropout:0.2,flatten,dense:64,dropout:0.5'
- 'conv2d:100,conv2d:64,maxpooling2d,dropout:0.5,flatten,dense:64,dense:24,dropout:0.2'
build:
image: python:3
build_steps:
- pip install -U -r requirements.txt
run:
cmd: export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib && \
python -u train.py --epochs={{ epochs }} \
--log_learning_rate={{ log_learning_rate }} \
--optimizer={{ optimizer }} \
--loss_metric={{ loss_metric }} \
--first_layer_output={{ first_layer_output }} \
--kernel_size={{ kernel_size }} \
--pool_size={{ pool_size }} \
--layers={{ layers }}