fix the batch size for update vertex

[Nicole00]

Please check the FAQ documentation before raising an issue

Describe the bug (required)
when the batch size of update vertex over 512, there's some assert error.

Your Environments (required)

• OS: uname -a
• Commit id (e.g. a3ffc7d8)

How To Reproduce(required)

Steps to reproduce the behavior:

1. Step 1
2. Step 2
3. Step 3

Expected behavior

Additional context