summaryrefslogtreecommitdiff
path: root/pyproject.toml
diff options
context:
space:
mode:
Diffstat (limited to 'pyproject.toml')
-rw-r--r--pyproject.toml5
1 files changed, 2 insertions, 3 deletions
diff --git a/pyproject.toml b/pyproject.toml
index 95e32a6..c977270 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,8 +22,8 @@ sphinx_rtd_theme = "^0.4.3"
boltons = "^20.1.0"
h5py = "^2.10.0"
toml = "^0.10.1"
-torch = "^1.6.0"
-torchvision = "^0.7.0"
+torch = "^1.7.0"
+torchvision = "^0.8.1"
loguru = "^0.5.0"
matplotlib = "^3.2.1"
tqdm = "^4.46.1"
@@ -37,7 +37,6 @@ python-Levenshtein = "^0.12.0"
defusedxml = "^0.6.0"
pytorch-block-sparse = "^0.1.2"
pytorch-metric-learning = "^0.9.92"
-faiss-gpu = "^1.6.3"
omegaconf = "^2.0.2"
[tool.poetry.dev-dependencies]