From 59cbf06e2ec67b28bfd46df8ae492d3bf149a764 Mon Sep 17 00:00:00 2001
From: tangwei12 <tangwei12@baidu.com>
Date: Tue, 11 Dec 2018 10:41:18 +0800
Subject: [PATCH] fix numel nce and prefetch

test=develop
---
 paddle/fluid/operators/nce_op.h | 1 -
 1 file changed, 1 deletion(-)

diff --git a/paddle/fluid/operators/nce_op.h b/paddle/fluid/operators/nce_op.h
index 99a3baba9..2c97eef09 100644
--- a/paddle/fluid/operators/nce_op.h
+++ b/paddle/fluid/operators/nce_op.h
@@ -49,7 +49,6 @@ void PrepareSamples(const framework::ExecutionContext &context,
   auto label = context.Input<Tensor>("Label");
   const int64_t *label_data = label->data<int64_t>();
   auto label_dims = label->dims();
-  //  int num_total_classes = context.Attr<int>("num_total_classes");
   // for unitest
   std::vector<int> custom_neg_classes =
       context.Attr<std::vector<int>>("custom_neg_classes");
-- 
GitLab