snark-hyper-example.yaml

version: 1
experiments:
  lr_search:
    image: tensorflow/tensorflow:1.11.0
    parameters:
      lr: '0.2-0.5'
    sampling: random # | grid | bayesian
    command:
      python train.py --lr {{lr}}

  arch_search:
    image: tensorflow/tensorflow:1.11.0
    parameters:
      lr: '0.2'
      arch: ['cnn', 'lstm', 'rnn']
    samples: 3
    command:
      python train.py --lr {{lr}} --arch {{arch}}