提交 7fa2e821 编写于 作者: T tangwei12

add local scope in nce

上级 627a6b8b
...@@ -194,6 +194,8 @@ class NCEKernel : public framework::OpKernel<T> { ...@@ -194,6 +194,8 @@ class NCEKernel : public framework::OpKernel<T> {
sample_out_data[i] += result(0); sample_out_data[i] += result(0);
sample_out_data[i] = (1. / (1. + exp(-sample_out_data[i]))); sample_out_data[i] = (1. / (1. + exp(-sample_out_data[i])));
} }
context.scope().DeleteScope(&local_scope);
#endif #endif
} else { } else {
auto weight_mat = auto weight_mat =
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册