fix batchsize bug