updates to how batching works in the hdfs mnist reader