Created by: mzchtx

I want to quantize a MobileNetV2 model with paddle-slim, the demo works fine. However, I get an error when adding a label_smooth layer before cross_entropy layer, the error information shows as the following picture.

The original demo is demo

And my modified code is：

#cost = fluid.layers.cross_entropy(input=out, label=label)

epsilon = 0.1
label_one_hot = fluid.layers.one_hot(input=label, depth=args.class_dim)
smooth_label = fluid.layers.label_smooth(label=label_one_hot, epsilon=epsilon, dtype="float32")
cost = fluid.layers.cross_entropy(input=out, label=smooth_label, soft_label=True)

• System: Centos 6.3
• Paddle: 1.4.1
• CUDA: 8.0
• CUDNN: v7