op auc does not have kernel for data_type[float]
Created by: 333caowei
本地cpu训练正常,gpu训练提示 op auc does not have kernel for data_type[float] 涉及代码如下:
binary_prediction = fluid.layers.concat(input=[1.0 - prediction, prediction], axis=1)
int_label = fluid.layers.cast(x=self.label, dtype='int64')
auc = fluid.layers.auc(input=binary_prediction, label=int_label)