昇腾故障案例详情页

训练服务器上做LLama13b模型推理时回复错误

更新时间: 2024/02/21

暂无评分

问题信息

问题来源产品大类产品子类关键字
官方模型推理PyTorch在线推理--

问题现象描述

  • 硬件配置:
    • Atlas800 9000 A2
    • mindformers套件
    • mindspore2.1
  • 问题现象:

    在Atlas800 9000 A2训练服务器上,利用Mindformers套件做Llama13B的推理时,推理结果不正确;

原因分析

推理精度不正确。

解决措施

在推理启动脚本中set_context时,设置precision_mode,具体描述参考mindspore官网链接

https://www.mindspore.cn/docs/zh-CN/r2.1/api_python/mindspore/mindspore.set_context.html?highlight=precision_mode

本页内容

该页面对您有帮助吗?
我要评分