Commit 9cffb74e authored by liucong's avatar liucong
Browse files

修改python示例部分代码

parent 3b905a5e
...@@ -37,7 +37,7 @@ while True: ...@@ -37,7 +37,7 @@ while True:
for _ in range(max_len): for _ in range(max_len):
# 推理 # 推理
result = model.run({inputName: migraphx.argument(input_ids)}) result = model.run({inputName: input_ids})
logits = [float(x) for x in result[0].tolist()] logits = [float(x) for x in result[0].tolist()]
# 对于[UNK]的概率设为无穷小,模型的预测结果不可能是[UNK] # 对于[UNK]的概率设为无穷小,模型的预测结果不可能是[UNK]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment