valohai.yaml

---

- step:
    name: preprocess data
    image: gcr.io/tensorflow/tensorflow:1.3.0-devel-gpu
    command: bash ./lib/preprocess.sh {parameters}
    inputs:
      - name: image_meta
        default: http://visualgenome.org/static/data/dataset/image_data.json.zip
      - name: regions
        default: http://visualgenome.org/static/data/dataset/region_descriptions.json.zip
    parameters:
      - name: vs
        type: float
        pass-as: -vs {v}
        default: 1.2
      - name: path
        type: string
        pass-as: -p {v}
        default: "/valohai/inputs"
      - name: output_dir
        type: string
        pass-as: -od {v}
        default: "/valohai/inputs/visual_genome"
      - name: max_words
        type: integer
        pass-as: -mw {v}
        default: 10

- step:
    name: download image data
    image: gcr.io/tensorflow/tensorflow:1.3.0-devel-gpu
    command: bash ./lib/download_data_vh.sh
    inputs:
      - name: image_1
        default: https://cs.stanford.edu/people/rak248/VG_100K_2/images.zip
      - name: image_2
        default: https://cs.stanford.edu/people/rak248/VG_100K_2/images2.zip

- step:
    name: train model
    image: gcr.io/tensorflow/tensorflow:1.3.0-devel-gpu
    command: bash ./tests/dencap_oa_test.sh {parameters}
    inputs:
      - name: vg_data
        default: ""
      - name: resnet
        default: https://drive.google.com/uc?export=download&confirm=aZtH&id=15PxiEp7HP-ZSBG9xHMamZr-zh8iBDeA4
      - name: image_1
        default: https://cs.stanford.edu/people/rak248/VG_100K_2/images.zip
      - name: image_2
        default: https://cs.stanford.edu/people/rak248/VG_100K_2/images2.zip
    parameters:
      - name: iters
        type: integer
        pass-as: -iters {v}
        default: 80000