{
  "pipeline": {
    "name": "train"
  },
  "transform": {
    "image": "tensor_flow_rnn",
    "cmd": [ "sh" ],
    "stdin": [
      "cd /code && python ptb_word_lm.py --data_path=/data/data --model small > /pfs/out/model-results" 
    ]
  },
  "parallelism": "1",
  "inputs": [
    {
      "repo": {
        "name": "tensor_ptb"
      },
      "method": "global"
    }
  ]
}

{
  "pipeline": {
    "name": "train"
  },
  "transform": {
    "image": "tensor_flow_rnn",
    "cmd": [ "sh" ],
    "stdin": [
      "cd /code && python ptb_word_lm.py --data_path=/data/data --model=test > /pfs/out/model-results" 
    ]
  },
  "parallelism": "1",
  "inputs": [
    {
      "repo": {
        "name": "tensor_ptb"
      },
      "method": "global"
    }
  ]
}

Use the test model so it runs quicker
Use the test model so it runs quicker
