diff --git a/requirements.txt b/requirements.txt index a295b23..88d3d05 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ opencv_python==4.5.1.48 numpy==1.19.3 torchvision==0.8.2 -torch==1.7.1 +torch==2.2.0 matplotlib==3.3.2 tensorboardX==2.2 scikit-image==0.17.2 \ No newline at end of file