diff --git a/requirements.txt b/requirements.txt index 24c584d..db7d287 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,5 +2,5 @@ h5py==2.9.0 matplotlib==3.1.1 tables==3.6.1 tb-nightly==2.2.0a20200212 -torch==1.4.0 +torch==2.2.0 torchvision==0.2.1 \ No newline at end of file