RTS / MLmodel
Ben Galewsky
Add example
d1eb2ce
artifact_path: model
flavors:
python_function:
config:
device: null
data: data
env:
conda: conda.yaml
virtualenv: python_env.yaml
loader_module: mlflow.pytorch
pickle_module_name: mlflow.pytorch.pickle_module
python_version: 3.10.11
pytorch:
code: null
model_data: data
pytorch_version: 2.5.0+cu124
mlflow_version: 2.17.0
model_size_bytes: 176239334
model_uuid: 23fc55bd7bea4ce0b7ea15d15fb19d4f
run_id: 5bc0ffd24c5748ba94f2adc6f33d83f6
signature:
inputs: '[{"type": "tensor", "tensor-spec": {"dtype": "float32", "shape": [1, 3,
226, 288]}}]'
outputs: '[{"name": "boxes", "type": "tensor", "tensor-spec": {"dtype": "float32",
"shape": [71, 4]}}, {"name": "labels", "type": "tensor", "tensor-spec": {"dtype":
"int64", "shape": [71]}}, {"name": "scores", "type": "tensor", "tensor-spec":
{"dtype": "float32", "shape": [71]}}, {"name": "masks", "type": "tensor", "tensor-spec":
{"dtype": "float32", "shape": [71, 1, 226, 288]}}]'
params: null
utc_time_created: '2024-10-23 22:18:33.740363'