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 }