Fix Torch Version Check for OSS Users

Summary: Pull Request resolved: https://github.com/facebookresearch/mobile-vision/pull/56

Reviewed By: ppwwyyxx

Differential Revision: D32576986

fbshipit-source-id: 1b20d1927a36ac80e33b51ff971b54767f647d43

d2go/export/api.py

@@ -51,7 +51,7 @@ from mobile_cv.predictor.builtin_functions import (
 )
 
 TORCH_VERSION: Tuple[int, ...] = tuple(int(x) for x in torch.__version__.split(".")[:2])
-if TORCH_VERSION >= (1, 10):
+if TORCH_VERSION > (1, 10):
     from torch.ao.quantization import convert
     from torch.ao.quantization.quantize_fx import convert_fx
 else:

d2go/modeling/quantization.py

@@ -19,7 +19,7 @@ from mobile_cv.arch.utils import fuse_utils
 from mobile_cv.common.misc.iter_utils import recursive_iterate
 
 TORCH_VERSION: Tuple[int, ...] = tuple(int(x) for x in torch.__version__.split(".")[:2])
-if TORCH_VERSION >= (1, 10):
+if TORCH_VERSION > (1, 10):
     from torch.ao.quantization.quantize_fx import convert_fx, prepare_fx, prepare_qat_fx
 else:
     from torch.quantization.quantize_fx import convert_fx, prepare_fx, prepare_qat_fx