Kevin Hu
Fix batch size issue. (#3675)
c9d78b3