update document

apache · Jan 24, 2019 · 70b1b2c · 70b1b2c
1 parent a451a96
commit 70b1b2c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/io/image_iter_common.h b/src/io/image_iter_common.h
@@ -127,7 +127,7 @@ struct ImageRecParserParam : public dmlc::Parameter<ImageRecParserParam> {
   int num_parts;
   /*! \brief the index of the part will read */
   int part_index;
-  /*! \brief device id on which to store the data */
+  /*! \brief device id used to create CPUPinned context for internal NDArray */
   int device_id;
   /*! \brief the size of a shuffle chunk */
   size_t shuffle_chunk_size;
@@ -166,7 +166,7 @@ struct ImageRecParserParam : public dmlc::Parameter<ImageRecParserParam> {
     DMLC_DECLARE_FIELD(part_index).set_default(0)
         .describe("The *i*-th virtual partition to be read.");
     DMLC_DECLARE_FIELD(device_id).set_default(0)
-        .describe("The device id on which to store the data.");
+        .describe("The device id used to create CPUPinned context for internal NDArray.");
     DMLC_DECLARE_FIELD(shuffle_chunk_size).set_default(0)
         .describe("The data shuffle buffer size in MB. Only valid if shuffle is true.");
     DMLC_DECLARE_FIELD(shuffle_chunk_seed).set_default(0)