From 70b1b2c0046d2d07c7aad51169d86f9f106d556e Mon Sep 17 00:00:00 2001 From: Yuxi Hu Date: Thu, 24 Jan 2019 14:33:24 -0800 Subject: [PATCH] update document --- src/io/image_iter_common.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/io/image_iter_common.h b/src/io/image_iter_common.h index b24e3bc0b529..99e6658d3d27 100644 --- a/src/io/image_iter_common.h +++ b/src/io/image_iter_common.h @@ -127,7 +127,7 @@ struct ImageRecParserParam : public dmlc::Parameter { 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 { 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)