diff --git a/requirements.txt b/requirements.txt index bad2311..349d496 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch==1.10.0 +torch==2.2.0 pytorch-lightning==1.5.1 torchmetrics==0.6.0 tqdm==4.62.3