File size: 936 Bytes
29c0409 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | name: "similarity_model"
backend: "python"
max_batch_size: 128
parameters: {
key: "EXECUTION_ENV_PATH",
value: {
# string_value: "/inspire/hdd/project/embodied-multimodality/public/cchang/env/audio.tar.gz"
# string_value: "/inspire/hdd/project/embodied-multimodality/public/cchang/env/audio_clean.tar.gz"
# string_value: "/inspire/hdd/project/embodied-multimodality/public/cchang/env_tar/audio_env.tar.gz"
string_value: "/inspire/hdd/project/embodied-multimodality/public/cchang/env/mooncast/bin/python"
}
}
input [
{
name: "AUDIO1"
data_type: TYPE_STRING
dims: [ 1 ] # 音频路径
},
{
name: "AUDIO2"
data_type: TYPE_STRING
dims: [ 1 ] # 音频路径
}
]
output [
{
name: "SIMILARITY"
data_type: TYPE_FP32
dims: [ 1 ] # 相似度分数
}
]
dynamic_batching {
preferred_batch_size: [ 16, 32 ]
}
instance_group [
{
count: 1
kind: KIND_GPU
}
]
|