[Fix] Fixed version comparison to include prerelease versions

[mattcasey02]

Motivation

Currently all tagged versions of torch 1.11.0 have version 1.11.0a0. Previously the comparison to 1.11.0 failed and self._sync_params() was still used, causing an error. This fix should include all versions of 1.11.

Modification

Changed a line of code to include prerelease versions in version checking.

BC-breaking (Optional)

N/A

Use cases (Optional)

N/A

Checklist

Before PR:

• I have read and followed the workflow indicated in the CONTRIBUTING.md to create this PR.
• Pre-commit or linting tools indicated in CONTRIBUTING.md are used to fix the potential lint issues.
• Bug fixes are covered by unit tests, the case that causes the bug should be added in the unit tests.
• New functionalities are covered by complete unit tests. If not, please add more unit test to ensure the correctness.
• The documentation has been modified accordingly, including docstring or example tutorials.

After PR:

• If the modification has potential influence on downstream or other related projects, this PR should be tested with some of those projects, like MMDet or MMCls.
• CLA has been signed and all committers have signed the CLA in this PR.

[CLAassistant]

All committers have signed the CLA.

[zhouzaida]

Hi @mattcasey02 , thanks for your contribution. Did you install pytorch build from source? I found the pre-build packages provided by torch are 1.11.0+cu113 or 1.11.0+cu115 which do not contain a0.

[mattcasey02]

Yes I installed from source. If you check out any of the tagged versions of torch 1.11 and look at the version.txt file they all have it listed as 1.11.0a0. For example, see here: https://github.com/pytorch/pytorch/blob/v1.11.0/version.txt

[zhouzaida]

Yes I installed from source. If you check out any of the tagged versions of torch 1.11 and look at the version.txt file they all have it listed as 1.11.0a0. For example, see here: https://github.com/pytorch/pytorch/blob/v1.11.0/version.txt

Got it. Thanks for your description.