mirror of
https://github.com/helm/charts.git
synced 2026-08-22 05:48:09 +00:00
Szymon Piotr KrasuskiandKubernetes Prow Robot
19b4b54a39
* Add custom variable to set number of trainign steps. By default script run in cheyang/distributed-tensorflow container runs with flag train_steps set to 0 which means that script will be running inifitely, However, it's to possible to set this flag to fixed number which will make script to run only passed number of training steps and stop. This change adds custom variable in hyperparams that lets user set custom value for training steps. Signed-off-by: Szymon <krasuski.szymon.piotr@gmail.com> * Bump chart version Signed-off-by: Szymon <krasuski.szymon.piotr@gmail.com>
26 lines
549 B
YAML
26 lines
549 B
YAML
# Default values for distributed-tensorflow.
|
|
# This is a YAML-formatted file.
|
|
# Declare variables to be passed into your templates.
|
|
worker:
|
|
number: 2
|
|
podManagementPolicy: Parallel
|
|
image:
|
|
repository: cheyang/distributed-tf
|
|
tag: 1.6.0
|
|
pullPolicy: IfNotPresent
|
|
port: 9000
|
|
# gpuCount: 2
|
|
ps:
|
|
number: 2
|
|
podManagementPolicy: Parallel
|
|
image:
|
|
repository: cheyang/distributed-tf
|
|
tag: 1.6.0
|
|
pullPolicy: IfNotPresent
|
|
port: 8000
|
|
# optimize for training
|
|
hyperparams:
|
|
batchsize: 20
|
|
learningrate: 0.001
|
|
trainsteps: 0
|