Skip to content

Commit 03e02cb

Browse files
authored
Update detectron2 and other requirements (#352)
* fix: pytorch version bug * feat: update detectron2 version
1 parent 4f436ae commit 03e02cb

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

detectron2

Submodule detectron2 updated 250 files

requirements.txt

+6-5
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,17 @@
11
cloudpickle
22
hydra-core
3-
omegaconf==2.1.0
3+
omegaconf
44
pybind11
5-
flake8==3.8.1
6-
isort==4.3.21
7-
black==22.3.0
5+
flake8
6+
isort
7+
black
88
autoflake
99
timm
1010
pytest
11-
scipy==1.7.3
11+
scipy
1212
psutil
1313
opencv-python
1414
wandb
1515
submitit
1616
einops
17+
fairscale

tests/utils/attention.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,7 @@
3737
from torch.nn.modules.module import Module
3838

3939
if (
40-
float(torch.__version__.split(".")[0]) == 0
41-
or (float(torch.__version__.split(".")[0]) == 1 and float(torch.__version__.split(".")[1])) < 7
40+
float(torch.__version__.split(".")[0]) == 0 or (float(torch.__version__.split(".")[0]) == 1 and float(torch.__version__.split(".")[1]) < 7)
4241
):
4342
from torch._overrides import has_torch_function, handle_torch_function
4443
else:
@@ -47,7 +46,7 @@
4746

4847
if (
4948
float(torch.__version__.split(".")[0]) == 0
50-
or (float(torch.__version__.split(".")[0]) == 1 and float(torch.__version__.split(".")[1])) < 9
49+
or (float(torch.__version__.split(".")[0]) == 1 and float(torch.__version__.split(".")[1]) < 9)
5150
):
5251
from torch.nn.modules.linear import _LinearWithBias
5352
else:

0 commit comments

Comments
 (0)