How to set batchsize and thread