add local scope in nce

6 years ago · 7fa2e821e4
parent 627a6b8bac
commit 7fa2e821e4
1 changed files with 2 additions and 0 deletions
--- 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 =