gistfile1.json

{
  "pipeline": {
    "name": "Mobile_Object_Train"
  },
  "transform": {
    "image": "nickharvey/detect-tf2:v2",
    "cmd": [ "sh" ],
    "stdin": [
      "python -m scripts.retrain --bottleneck_dir=/pfs/out/bottlenecks --how_many_training_steps=500 --model_dir=/pfs/out/models/ --summaries_dir=/pfs/out/training_summaries/${ARCHITECTURE} --output_graph=/pfs/out/retrained_graph.pb --output_labels=/pfs/out/retrained_labels.txt --architecture=${ARCHITECTURE} --image_dir=/pfs/flower_photos/",
      "tensorboard --logdir /pfs/out/training_summaries &"
    ],
    "env": {
      "IMAGE_SIZE": "224",
      "ARCHITECTURE": "mobilenet_0.50_${IMAGE_SIZE}"
    }
  },
  "parallelism_spec": {
       "constant" : 1
  },
  "service": {
    "internal_port": 6006,
    "external_port": 30006
  },
  "input": {
    "pfs": {
      "repo": "flower_photos",
      "glob": "/"
    }
  }
}