This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
USE_GPU = True | |
FIL_MODEL_DIR = "./model_repository/fil" | |
# Maximum size in bytes for input and output arrays. If you are | |
# using Triton 21.11 or higher, all memory allocations will make | |
# use of Triton's memory pool, which has a default size of | |
# 67_108_864 bytes | |
MAX_MEMORY_BYTES = 60_000_000 | |
NUM_FEATURES = 15 | |
NUM_CLASSES = 2 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/bin/bash | |
conda create -y -n preprocessing_env python=3.8 | |
source /opt/conda/etc/profile.d/conda.sh | |
conda activate preprocessing_env | |
export PYTHONNOUSERSITE=True | |
conda install -y -c conda-forge pandas scikit-learn | |
pip install conda-pack | |
conda-pack |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: "preprocessing" | |
backend: "python" | |
max_batch_size: 882352 | |
input [ | |
{ | |
name: "User" | |
data_type: TYPE_FP32 | |
dims: [ 1 ] | |
}, | |
{ |