diff --git a/RobustParTAK4/PUPPI/V00/modelfile b/RobustParTAK4/PUPPI/V00/modelfile new file mode 120000 index 0000000..7f65391 --- /dev/null +++ b/RobustParTAK4/PUPPI/V00/modelfile @@ -0,0 +1 @@ +../../../models/particletransformer_AK4/1/ \ No newline at end of file diff --git a/RobustParTAK4/PUPPI/V00/RobustParTAK4.onnx b/models/particletransformer_AK4/1/model.onnx similarity index 100% rename from RobustParTAK4/PUPPI/V00/RobustParTAK4.onnx rename to models/particletransformer_AK4/1/model.onnx diff --git a/models/particletransformer_AK4/config.pbtxt b/models/particletransformer_AK4/config.pbtxt new file mode 100644 index 0000000..9c0e42e --- /dev/null +++ b/models/particletransformer_AK4/config.pbtxt @@ -0,0 +1,56 @@ +name: "particletransformer_AK4" +platform: "onnxruntime_onnx" +max_batch_size: 500 +input { + name: "input_1" + data_type: TYPE_FP32 + dims: [-1, 16] +} +input { + name: "input_2" + data_type: TYPE_FP32 + dims: [-1, 8] +} +input { + name: "input_3" + data_type: TYPE_FP32 + dims: [-1, 14] +} +input { + name: "input_4" + data_type: TYPE_FP32 + dims: [-1, 4] +} +input { + name: "input_5" + data_type: TYPE_FP32 + dims: [-1, 4] +} +input { + name: "input_6" + data_type: TYPE_FP32 + dims: [-1, 4] +} +output { + name: "softmax" + data_type: TYPE_FP32 + dims: [6] + label_filename: "particletransformer_labels.txt" +} +dynamic_batching { + preferred_batch_size: [200] +} +optimization { + graph { + level: -1 + } +} +model_repository_agents { + agents { + name: "checksum" + parameters { + key: "MD5:1/model.onnx" + value: "e9ad0c98366e337e2562a9522ffd575b" + } + } +} diff --git a/models/particletransformer_AK4/particletransformer_labels.txt b/models/particletransformer_AK4/particletransformer_labels.txt new file mode 100644 index 0000000..e4068c0 --- /dev/null +++ b/models/particletransformer_AK4/particletransformer_labels.txt @@ -0,0 +1,6 @@ +b +bb +lepb +c +uds +g