diff --git a/requirements.txt b/requirements.txt index 1e512fb..42eca48 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ tensorboardX==1.9 -torch==1.6.0 +torch==2.2.0 torchvision==0.7.0 sklearn==0.0 pandas==0.25.2