towhee.serve.triton.to_triton_models.ModelToTriton

class towhee.serve.triton.to_triton_models.ModelToTriton(op, model_root, model_name, op_config)[source]

Bases: ToTriton

NNOp to triton model.

Convert model to trt, torchscript or onnx.

Methods

to_triton

Attributes

inputs

outputs

__init__(op, model_root, model_name, op_config)[source]