A kind suggestion on the pytorch version.

[guanchuwang]

This is a kind suggestion to the authors to revise the required pytorch version to "torch >= 1.7.1" and "torch <= 1.8.0" in the readme file.

[merrymercy]

Thanks for the suggestion.
It is updated 370bc9f

You can also find a possible fix in this PR #22