Fix PyTorch version check
This commit is contained in:
parent
e184f608c5
commit
c7be85d4e7
|
@ -19,7 +19,7 @@ except ImportError:
|
||||||
BICUBIC = Image.BICUBIC
|
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")
|
warnings.warn("PyTorch version 1.7.1 or higher is recommended")
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue