blob: 6283159c324c9cd6229df89f9249b378b967e765 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
model[].name "my_ranking_model"
model[].fileref "files/ranking_model.onnx"
model[].input[].name "input_ids"
model[].input[].source "rankingExpression(input_ids)"
model[].input[].name "attention_mask"
model[].input[].source "rankingExpression(attention_mask)"
model[].input[].name "token_type_ids"
model[].input[].source "rankingExpression(token_type_ids)"
model[].output[].name "score"
model[].output[].as "score"
model[].dry_run_on_setup true
model[].stateless_execution_mode ""
model[].stateless_interop_threads -1
model[].stateless_intraop_threads -1
model[].gpu_device -1
model[].gpu_device_required false
|
