[Fix] Reverting back to torch==1.9.0

setup.py

@@ -56,12 +56,12 @@
     f"{'deepsparse' if is_release else 'deepsparse-nightly'}~={version_nm_deps}"
 ]
 _pytorch_deps = [
-    "torch>=1.1.0,<=1.10.2",
+    "torch>=1.1.0,<=1.9.1",
     "tensorboard>=1.0",
     "tensorboardX>=1.0",
     "gputils",
 ]
-_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<=0.11.3"]
+_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<=0.10.1"]
 _tensorflow_v1_deps = ["tensorflow<2.0.0", "tensorboard<2.0.0", "tf2onnx>=1.0.0,<1.6"]
 _tensorflow_v1_gpu_deps = [
     "tensorflow-gpu<2.0.0",

src/sparseml/pytorch/base.py

@@ -49,7 +49,7 @@
 
 
 _TORCH_MIN_VERSION = "1.0.0"
-_TORCH_MAX_VERSION = "1.10.100"  # set bug to 100 to support all future 1.10.X  versions
+_TORCH_MAX_VERSION = "1.9.100"  # set bug to 100 to support all future 1.9.X versions
 
 
 def check_torch_install(