diff --git a/pyproject.toml b/pyproject.toml index 6440af7b..8be2788b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,10 +11,10 @@ authors = [ requires-python = ">=3.9" dependencies = [ - "torch ~= 1.13.1", - "torchvision ~= 0.14.1", - "pytorch-lightning ~= 1.6.5", - "torchmetrics == 0.6.0", + "torch ~= 2.0.1", + "torchvision ~= 0.15.2", + "pytorch-lightning ~= 2.0.2", + "torchmetrics == 0.11.4", # --------- hydra --------- # "hydra-core ~= 1.2.0",