Update detectron2 and other requirements (#352)

* fix: pytorch version bug

* feat: update detectron2 version

requirements.txt

@@ -1,16 +1,17 @@
 cloudpickle
 hydra-core
-omegaconf==2.1.0
+omegaconf
 pybind11
-flake8==3.8.1 
-isort==4.3.21
-black==22.3.0
+flake8 
+isort
+black
 autoflake
 timm
 pytest
-scipy==1.7.3
+scipy
 psutil
 opencv-python
 wandb
 submitit
 einops
+fairscale

tests/utils/attention.py

@@ -37,8 +37,7 @@
 from torch.nn.modules.module import Module
 
 if (
-    float(torch.__version__.split(".")[0]) == 0
-    or (float(torch.__version__.split(".")[0]) == 1 and float(torch.__version__.split(".")[1])) < 7
+    float(torch.__version__.split(".")[0]) == 0 or (float(torch.__version__.split(".")[0]) == 1 and float(torch.__version__.split(".")[1]) < 7)
 ):
     from torch._overrides import has_torch_function, handle_torch_function
 else:
@@ -47,7 +46,7 @@
 
 if (
     float(torch.__version__.split(".")[0]) == 0
-    or (float(torch.__version__.split(".")[0]) == 1 and float(torch.__version__.split(".")[1])) < 9
+    or (float(torch.__version__.split(".")[0]) == 1 and float(torch.__version__.split(".")[1]) < 9) 
 ):
     from torch.nn.modules.linear import _LinearWithBias
 else: