Exemplos de create_identity_ensemble_modelfile em Python

Exemplo n.º 1

Arquivo: gen_qa_identity_models.py Projeto: zjwzcnjsy/tensorrt-inference-server

def create_ensemble_modelfile(create_savedmodel, models_dir, model_version,
                              io_cnt, max_batch, dtype, shape):
    if not tu.validate_for_ensemble_model("zero", dtype, dtype, dtype, shape,
                                          shape, shape):
        return

    emu.create_identity_ensemble_modelfile("zero", models_dir, model_version,
                                           max_batch, dtype, [shape] * io_cnt,
                                           [shape] * io_cnt)

Exemplo n.º 2

Arquivo: gen_qa_reshape_models.py Projeto: rotorliu/triton-inference-server

def create_ensemble_modelfile(models_dir, model_version, max_batch, dtype,
                              input_shapes, output_shapes):

    assert len(input_shapes) == len(output_shapes)
    if not tu.validate_for_ensemble_model("reshape", dtype, dtype, dtype,
                                          input_shapes[0], input_shapes[0],
                                          input_shapes[0]):
        return

    emu.create_identity_ensemble_modelfile("reshape", models_dir, model_version,
                                           max_batch, dtype, input_shapes,
                                           output_shapes)