[BUGFIX] fix a SQUAD2.0 bug

[eric-haibin-lin]

Description

(Brief description on what this PR is about)

Checklist

Essentials

• PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc)
• Changes are complete (i.e. I finished coding on this PR)
• All changes have test coverage
• Code is well-documented

Changes

• Feature1, tests, (and when applicable, API doc)
• Feature2, tests, (and when applicable, API doc)

Comments

• If this change is a backward incompatible change, why must this change be made.
• Interesting edge cases to note here

[codecov]

Codecov Report

❗ No coverage uploaded for pull request head (sq@508e60e). Click here to learn what that means.
The diff coverage is n/a.

[codecov]

Codecov Report

❗ No coverage uploaded for pull request head (sq@508e60e). Click here to learn what that means.
The diff coverage is n/a.

[mli]

Job PR-907/1 is complete.
Docs are uploaded to http://gluon-nlp-staging.s3-accelerate.dualstack.amazonaws.com/PR-907/1/index.html

[szha]

Is this still WIP?

[mli]

Job PR-907/11 is complete.
Docs are uploaded to http://gluon-nlp-staging.s3-accelerate.dualstack.amazonaws.com/PR-907/11/index.html