edugp's picture Fix batch sizes 0af32aa