guild.yml

train:
  description: Train the AMAZING model
  main: train
  flags:
    $dest: globals
    epochs:
      description: Number of epochs to train
      default: 6
      min: 1
      max: 10
    activation:
      description: Activation function
      default: relu
      choices:
        - relu
        - sigmoid
    num_dense_layers:
      description: Number of dense layers
      default: 1
      min: 1
      max: 5
    num_dense_nodes:
      description: Number of nodes per dense layer
      default: 100
      min: 5
      max: 512
    learning_rate:
      description: Learning rate
      default: 1e-3
      min: 1e-6
      max: 1e-1
      distribution: loguniform
    no_early_stopping:
      description: Set to 'yes' to bypass early stopping
      flag-value: yes
      null-label: stop early as needed
  objective:
    maximize: epoch_val_acc
  compare:
    - =activation
    - =learning_rate
    - =num_dense_layers
    - =num_dense_nodes
    - =epochs
    - epoch_loss step as step
    - epoch_loss as loss
    - epoch_val_acc as acc
  requires:
    path: data
    default-unpack: no
    sources:
      - url: http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz
      - url: http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz
      - url: http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz
      - url: http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz

reproduce-result:
  description: Reproduce published result
  default: true
  steps:
    - train epochs=15 no_early_stopping=yes --label 'published result'