From d52fd00a66c6278eb718d378ea67226cdb1633b6 Mon Sep 17 00:00:00 2001
From: sweetsky0901 <work@yq01-idl-gpu-online20.yq01.baidu.com>
Date: Mon, 1 Jan 2018 22:11:17 +0800
Subject: [PATCH] int to size_t

---
 paddle/operators/detection_output_op.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/operators/detection_output_op.h b/paddle/operators/detection_output_op.h
index cd6417087a..f8abd5b640 100644
--- a/paddle/operators/detection_output_op.h
+++ b/paddle/operators/detection_output_op.h
@@ -63,7 +63,7 @@ class DetectionOutputKernel : public framework::OpKernel<T> {
     float nms_threshold = context.template Attr<float>("nms_threshold");
     float confidence_threshold =
         context.template Attr<float>("confidence_threshold");
-    int batch_size = in_conf->dims()[1];
+    size_t batch_size = in_conf->dims()[1];
     int conf_sum_size = in_conf->numel();
     // for softmax
     std::vector<int64_t> conf_shape_softmax_vec(