diff --git a/requirements.txt b/requirements.txt index fe60483..01101aa 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ pytorch-lightning==0.7.6 scikit-learn==0.23.1 scipy==1.4.1 tensorboard==2.2.2 -torch==1.5.0 +torch==2.2.0 torchvision==0.6.0a0+82fd1c8 tqdm==4.46.1 transformers==2.11.0 \ No newline at end of file