Enable device automatically switching in mine_hard_examples_op. (#8706)

7 years ago · 457d012a8e
parent bbf98a0176
commit 457d012a8e
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/mine_hard_examples_op.cc
+++ b/paddle/fluid/operators/mine_hard_examples_op.cc
@ -247,7 +247,7 @@ class MineHardExamplesOp : public framework::OperatorWithKernel {
      const framework::ExecutionContext& ctx) const override {
    return framework::OpKernelType(
        framework::ToDataType(ctx.Input<framework::Tensor>("ClsLoss")->type()),
-        ctx.device_context());
+        platform::CPUPlace());
  }
 };