diff --git a/requirements.txt b/requirements.txt index 63c3713..9c42c33 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,6 +3,6 @@ h5py==2.8.0 numpy==1.14.5 Pillow>=7.1.0 scipy>=1.0.0 -torch==1.2.0 +torch==2.2.0 torchvision==0.2.1 tensorboardX \ No newline at end of file