diff --git a/requirements.txt b/requirements.txt index 8663c2b..5ff742f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ h5py nltk numpy Pillow -torch==1.2.0 +torch==2.2.0 torchvision==0.2.1 tqdm termcolor