ARTICLE · INTELLIGENCE

战地情报 · 详情页

来自尧图项目组的一线实战观察与深度解析

105、在昇腾910C上适配chinese-roberta-wwm-ext

105、在昇腾910C上适配chinese-roberta-wwm-ext

基本思想:在昇腾910C上适配chinese-roberta-wwm-ext

镜像地址:crpi-8ew3ouqcvy9yujug.cn-hangzhou.personal.cr.aliyuncs.com/sxj731533730/chinese-roberta-wwm-ext_910c:v0

一、下载模型

[root@hostname-g6ilx data]# cat down.py #模型下载 from modelscope import snapshot_download model_dir = snapshot_download('dienstag/chinese-roberta-wwm-ext',cache_dir=".") [root@hostname-g6ilx data]#

设备类型

二、模型转onnx

import os import shutil import numpy as np import torch from transformers import AutoModelForMaskedLM, AutoTokenizer # ---- 路径配置 ------------------------------------------------------------- HERE = os.path.dirname(os.path.abspath(__
RELATED READING

延伸阅读

更多一线实战笔记与深度复盘,助您持续精进