pere's picture
batch size 64 supported
ee26a32