[GitHub] chinakook commented on a change in pull request #12285: [WIP] Change the way NDArrayIter handle the last batch

[GitBox <gi...@apache.org>]

chinakook commented on a change in pull request #12285: [WIP] Change the way NDArrayIter handle the last batch
URL: https://github.com/apache/incubator-mxnet/pull/12285#discussion_r212159035
 
 

 ##########
 File path: python/mxnet/io.py
 ##########
 @@ -633,8 +631,11 @@ class NDArrayIter(DataIter):
         Only supported if no h5py.Dataset inputs are used.
     last_batch_handle : str, optional
         How to handle the last batch. This parameter can be 'pad', 'discard' or
-        'roll_over'. 'roll_over' is intended for training and can cause problems
-        if used for prediction.
+        'roll_over'.
+        If 'pad', the last batch will be padded with data starting from the begining
 
 Review comment:
   Yeah, It's so clear with a example.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services