Skip to content

Instantly share code, notes, and snippets.

@rags1357
Created February 13, 2023 23:50
Show Gist options
  • Save rags1357/f37e72b19eb83a0b9b01738ca55783af to your computer and use it in GitHub Desktop.
Save rags1357/f37e72b19eb83a0b9b01738ca55783af to your computer and use it in GitHub Desktop.
config.pbtxt
name: "preprocessing"
backend: "python"
max_batch_size: 882352
input [
{
name: "User"
data_type: TYPE_FP32
dims: [ 1 ]
},
{
name: "Card"
data_type: TYPE_FP32
dims: [ 1 ]
},
{
name: "Year"
data_type: TYPE_FP32
dims: [ 1 ]
},
{
name: "Month"
data_type: TYPE_FP32
dims: [ 1 ]
},
{
name: "Day"
data_type: TYPE_FP32
dims: [ 1 ]
},
{
name: "Time"
data_type: TYPE_STRING
dims: [ 1 ]
},
{
name: "Amount"
data_type: TYPE_STRING
dims: [ 1 ]
},
{
name: "Use Chip"
data_type: TYPE_STRING
dims: [ 1 ]
},
{
name: "Merchant Name"
data_type: TYPE_STRING
dims: [ 1 ]
},
{
name: "Merchant City"
data_type: TYPE_STRING
dims: [ 1 ]
},
{
name: "Merchant State"
data_type: TYPE_STRING
dims: [ 1 ]
},
{
name: "Zip"
data_type: TYPE_STRING
dims: [ 1 ]
},
{
name: "MCC"
data_type: TYPE_STRING
dims: [ 1 ]
},
{
name: "Errors?"
data_type: TYPE_STRING
dims: [ 1 ]
}
]
output [
{
name: "OUTPUT"
data_type: TYPE_FP32
dims: [ 15 ]
}
]
instance_group [
{
count: 1
kind: KIND_CPU
}
]
parameters: {
key: "EXECUTION_ENV_PATH",
value: {string_value: "$$TRITON_MODEL_DIRECTORY/preprocessing_env.tar.gz"}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment