PIE
PIE copied to clipboard
Fix a minor bug.
The original code produces an additional redundant batch when the number of data samples is divisible by the batch size.