diff options
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/pyproject.toml b/pyproject.toml index 6c5a2a0..7d81365 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -15,8 +15,6 @@ click = "^7.1.2" boltons = "^20.1.0" h5py = "^3.2.1" toml = "^0.10.1" -torch = "^1.9.0" -torchvision = "^0.10.0" loguru = "^0.5.0" matplotlib = "^3.2.1" tqdm = "^4.46.1" @@ -24,7 +22,7 @@ opencv-python = "^4.3.0" nltk = "^3.5" torch-summary = "^1.4.2" defusedxml = "^0.6.0" -omegaconf = "^2.0.2" +omegaconf = "^2.1.0" einops = "^0.3.0" gtn = "^0.0.0" sentencepiece = "^0.1.95" @@ -33,8 +31,10 @@ Pillow = "^8.1.2" madgrad = "^1.0" editdistance = "^0.5.3" torchmetrics = "^0.4.1" -hydra-core = "^1.0.6" +hydra-core = "^1.1.0" attr = "^0.3.1" +torch = "^1.9.0" +torchvision = "^0.10.0" [tool.poetry.dev-dependencies] pytest = "^5.4.2" @@ -50,7 +50,7 @@ flake8-import-order = "^0.18.1" safety = "^1.9.0" mypy = "^0.770" typeguard = "^2.7.1" -wandb = "^0.10.30" +wandb = "^0.11.2" scipy = "^1.6.1" flake8-annotations = "^2.6.2" flake8-docstrings = "^1.6.0" |