diff --git a/requirements.txt b/requirements.txt index b448581..92baaa8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,5 +5,5 @@ pretrainedmodels==0.7.4 python-dateutil==2.8.0 pytz==2019.2 six==1.12.0 -torch==1.5.0 +torch==2.2.0 torchvision==0.6.0