#!/bin/sh

set -e

python onnx_export.py \
  --config-path=/model/config.json \
  --model-path=/model/model.pt \
  export /tmp/model.onnx /tmp/model_one_step.onnx

python trt_builder.py \
  --config_path=/model/config.json \
  build-trt /tmp/model.onnx /model/model.trt

python trt_builder.py \
  --config_path=/model/config.json \
  build-trt-one-step /tmp/model_one_step.onnx /model/model_one_step.trt