diff --git a/requirements.txt b/requirements.txt index 14b1dbc..bee0d1e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ numpy==1.22.0 pandas==1.3.4 scikit_learn==1.0.1 simple_parsing==0.0.17 -torch==1.10.0 +torch==2.2.0 torchvision==0.11.1 tqdm==4.62.3 wandb==0.12.7