首页 PaddleNLP 帖子详情
新闻文本分类问题
收藏
快速回复
PaddleNLP 问答NLP 2587 4
新闻文本分类问题
收藏
快速回复
PaddleNLP 问答NLP 2587 4

一个使用CNN进行文本分类的例子,链接如下:

https://aistudio.baidu.com/bdvgpu/user/83109/185742/notebooks/185742.ipynb?redirects=1

在AI Studio上可以正常运行

但是在本地运行时,训练、存储模型等都可以成功,但是在预测模块出现以下问题

PaddleCheckError: Tensor holds the wrong type, it holds int, but desires to be int64_t at [D:\1.6.1\paddle\paddle/fluid/framework/tensor_impl.h:30]

我直接用load_inference_model读取的模型,为什么跟存储的模型不一样。。。

[name: "save_infer_model/scale_0"
type {
type: LOD_TENSOR
lod_tensor {
tensor {
data_type: FP32
dims: -1
dims: 10
}
lod_level: 0
}
}

这里的int64为什么是FP32?

0
收藏
回复
全部评论(4)
时间顺序
8
8melody8
#2 回复于2019-11

我也遇到了同样的问题,不知道如何解决

0
回复
8
8melody8
#3 回复于2019-11

问题解决了,get_data()函数的返回值要做一下处理

def get_data(sentence):
    with open(dict_path, 'r', encoding='utf-8') as f_data:
        dict_txt = eval(f_data.readlines()[0])
    
    dict_txt = dict(dict_txt)
    
    keys = dict_txt.keys()
    
    data = []
    
    for s in sentence:
        if not s in keys:
            s = ''
        
        data.append(int(dict_txt[s]))
    
    return np.array(data, dtype = np.int64)
2
回复
nk_ganggang
#4 回复于2019-11
问题解决了,get_data()函数的返回值要做一下处理 [代码]

真是太感谢了,我马上去试试~~~~

1
回复
G
GoodmanWZ
#5 回复于2019-12

请问是怎么解决的!求告知

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户