Fix PyTorch version check
This commit is contained in:
parent
e184f608c5
commit
c7be85d4e7
|
@ -19,7 +19,7 @@ except ImportError:
|
|||
BICUBIC = Image.BICUBIC
|
||||
|
||||
|
||||
if torch.__version__.split(".") < ["1", "7", "1"]:
|
||||
if [int(n) for n in torch.__version__.split(".")] < [1, 7, 1]:
|
||||
warnings.warn("PyTorch version 1.7.1 or higher is recommended")
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue