首页 Paddle框架 帖子详情
python2.7paddle1.6下,embedding函数的dtype参数若采用unicode会报错。
收藏
快速回复
Paddle框架 问答深度学习 15915 1
python2.7paddle1.6下,embedding函数的dtype参数若采用unicode会报错。
收藏
快速回复
Paddle框架 问答深度学习 15915 1

测试环境: python2.7 + paddle 1.6.0
embedding函数的dtype参数若采用unicode会报错
复现代码:

from __future__ import unicode_literals
import paddle as p
import paddle.fluid as f
from paddle.fluid import core


print(p.__version__)
ids = f.data(shape=[None, None, 1], dtype='int64', name='a')
f.layers.embedding(ids, size=[10000, 32], param_attr=f.ParamAttr(name='emb'), dtype='float32')

错误log

1.6.0
Traceback (most recent call last):
  File "shit17.py", line 9, in <module>
    f.layers.embedding(ids, size=[10000, 32], param_attr=f.ParamAttr(name='emb'), dtype='float32')
  File "/home/work/chenxuyi/playground/off-ernie/app/lib/python2.7/site-packages/paddle/fluid/layers/nn.py", line 614, in embedding
    if convert_dtype(dtype) in ['float16']:
  File "/home/work/chenxuyi/playground/off-ernie/app/lib/python2.7/site-packages/paddle/fluid/data_feeder.py", line 37, in convert_dtype
    if dtype == core.VarDesc.VarType.BOOL:
TypeError: __eq__(): incompatible function arguments. The following argument types are supported:
    1. (self: paddle.fluid.core_avx.VarDesc.VarType, arg0: paddle.fluid.core_avx.VarDesc.VarType) -> bool
    2. (self: paddle.fluid.core_avx.VarDesc.VarType, arg0: int) -> bool

Invoked with: VarType.BOOL, u'float32
0
收藏
回复
全部评论(1)
时间顺序
AIStudio792103
#2 回复于2019-11

你好,可以更新到1.6.1版本试下

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