首页 Paddle框架 帖子详情
fluid.layers.data_norm 使用报错
收藏
快速回复
Paddle框架 问答深度学习 997 1
fluid.layers.data_norm 使用报错
收藏
快速回复
Paddle框架 问答深度学习 997 1

在静态图中对variable矩阵使用fluid.layers.data_norm归一化,报如下问题

Error Message Summary:

PaddleCheckError: op data_norm does not have kernel for data_type[float]:data_layout[ANY_LAYOUT]:place[CUDAPlace(0)]:library_type[PLAIN] at [/paddle/paddle/fluid/framework/operator.cc:1010]
[operator < data_norm > error]

定义语句为:
weight = fluid.layers.create_parameter(shape=[self.in_dim, self.out_dim],
dtype='float32', default_initializer=fluid.initializer.Xavier(uniform=True))

    weight_norm = fluid.layers.data_norm(weight, name='weight')

去掉该归一化步骤可以正常运行,请帮忙看下该op的问题如何解决

0
收藏
回复
全部评论(1)
时间顺序
hutuxian
#2 回复于2019-12

请您提供一下所使用的paddle版本?
data_norm op在1.6.1之前是没有CUDA版本的。预计会在下一个major版本提供CUDA实现。

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