mirror of
https://github.com/hanzo-ml/katib.git
synced 2026-09-24 00:31:27 +00:00
Mahdi KhashanandGitHub
19db2d47d4
* move to ghcr Signed-off-by: mahdikhashan <[email protected]> * move images to ghcr Signed-off-by: mahdikhashan <[email protected]> * manifests Signed-off-by: mahdikhashan <[email protected]> * change registry in all path Signed-off-by: mahdikhashan <[email protected]> * update script Signed-off-by: mahdikhashan <[email protected]> * fix Signed-off-by: mahdikhashan <[email protected]> * fix Signed-off-by: mahdikhashan <[email protected]> * slight fix Signed-off-by: mahdikhashan <[email protected]> --------- Signed-off-by: mahdikhashan <[email protected]> Signed-off-by: Mahdi Khashan <[email protected]>
71 lines
1.9 KiB
YAML
71 lines
1.9 KiB
YAML
---
|
|
apiVersion: kubeflow.org/v1beta1
|
|
kind: Experiment
|
|
metadata:
|
|
namespace: kubeflow
|
|
name: darts-cpu
|
|
spec:
|
|
parallelTrialCount: 1
|
|
maxTrialCount: 1
|
|
maxFailedTrialCount: 1
|
|
objective:
|
|
type: maximize
|
|
objectiveMetricName: Best-Genotype
|
|
metricsCollectorSpec:
|
|
collector:
|
|
kind: StdOut
|
|
source:
|
|
filter:
|
|
metricsFormat:
|
|
- "([\\w-]+)=(Genotype.*)"
|
|
algorithm:
|
|
algorithmName: darts
|
|
algorithmSettings:
|
|
- name: num_epochs
|
|
value: "1"
|
|
- name: num_nodes
|
|
value: "1"
|
|
- name: init_channels
|
|
value: "1"
|
|
- name: stem_multiplier
|
|
value: "1"
|
|
nasConfig:
|
|
graphConfig:
|
|
numLayers: 1
|
|
operations:
|
|
- operationType: max_pooling
|
|
parameters:
|
|
- name: filter_size
|
|
parameterType: categorical
|
|
feasibleSpace:
|
|
list:
|
|
- "3"
|
|
trialTemplate:
|
|
primaryContainerName: training-container
|
|
trialParameters:
|
|
- name: algorithmSettings
|
|
description: Algorithm settings of DARTS Experiment
|
|
reference: algorithm-settings
|
|
- name: searchSpace
|
|
description: Search Space of DARTS Experiment
|
|
reference: search-space
|
|
- name: numberLayers
|
|
description: Number of Neural Network layers
|
|
reference: num-layers
|
|
trialSpec:
|
|
apiVersion: batch/v1
|
|
kind: Job
|
|
spec:
|
|
template:
|
|
spec:
|
|
containers:
|
|
- name: training-container
|
|
image: ghcr.io/kubeflow/katib/darts-cnn-cifar10-cpu:latest
|
|
command:
|
|
- python3
|
|
- run_trial.py
|
|
- --algorithm-settings="${trialParameters.algorithmSettings}"
|
|
- --search-space="${trialParameters.searchSpace}"
|
|
- --num-layers="${trialParameters.numberLayers}"
|
|
restartPolicy: Never
|