set batch size to 128 + heap size to 8Gb