add local scope in nce

7fa2e821 · tangwei12 · 627a6b8b · 7fa2e821
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/operators/nce_op.h paddle/fluid/operators/nce_op.h +2 -0

未找到文件。
--- a/paddle/fluid/operators/nce_op.h
+++ b/paddle/fluid/operators/nce_op.h
@@ -194,6 +194,8 @@ class NCEKernel : public framework::OpKernel<T> {
        sample_out_data[i] += result(0);
        sample_out_data[i] = (1. / (1. + exp(-sample_out_data[i])));
      }
+
+      context.scope().DeleteScope(&local_scope);
 #endif
    } else {
      auto weight_mat =