diff --git a/requirements.txt b/requirements.txt index 4d0ad6eaa..d06a91b71 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,13 +1,13 @@ # --------- pytorch --------- # -# torch>=2.0.0 -# torchvision>=0.15.0 -# lightning>=2.0.0 -# torchmetrics>=0.11.4 +torch>=2.0.0 +torchvision>=0.15.0 +lightning>=2.0.0 +torchmetrics>=0.11.4 # --------- hydra --------- # -# hydra-core==1.3.2 -# hydra-colorlog==1.2.0 -# hydra-optuna-sweeper==1.2.0 +hydra-core==1.3.2 +hydra-colorlog==1.2.0 +hydra-optuna-sweeper==1.2.0 # --------- loggers --------- # # wandb