the number of test samples does not have to be the exact multiple of batch_size