github.com/pachyderm/pachyderm@v1.13.4/examples/ml/neon/infer.json (about)

     1  {
     2    "pipeline": {
     3      "name": "inference"
     4    },
     5    "description": "An inference pipeline that makes a prediction based on the trained model and the data in the `reviews` repo.",
     6    "transform": {
     7      "image": "dwhitena/neon-inference",
     8      "cmd": [
     9        "python",
    10        "examples/imdb/auto_inference.py",
    11        "--model_weights",
    12        "/pfs/model/imdb.p",
    13        "--vocab_file",
    14        "/pfs/model/imdb.vocab",
    15        "--review_files",
    16        "/pfs/reviews",
    17        "--output_dir",
    18        "/pfs/out"
    19      ]
    20    },
    21    "parallelism_spec": {
    22      "constant": "2"
    23    },
    24    "input": {
    25      "cross": [
    26        {
    27          "pfs": {
    28            "repo": "reviews",
    29            "glob": "/*"
    30          }
    31        },
    32        {
    33          "pfs": {
    34            "repo": "model",
    35            "glob": "/"
    36          }
    37        }
    38      ]
    39    }
    40  }