search

taishan1994 / pytorch_bert_bilstm_crf_ner

基于pytorch的bert_bilstm_crf中文命名实体识别
517 stars 81 forks source link

在执行convert_onnx.py时出现异常 #61

Closed tjulh closed 6 months ago

tjulh commented 6 months ago

作者你好，我在执行你的convert_onnx.py将模型转为onnx时，有warning如下 图片 生成的onnx文件很小，也无法加载推理

taishan1994 commented 6 months ago

把torch.tensor()去掉看看

tjulh commented 6 months ago

把torch.tensor()去掉看看

解决了，看到你的readme里面有说CRF需要换掉