diff --git a/pip-requirements.txt b/pip-requirements.txt index 7eaea4f..2a8cee2 100644 --- a/pip-requirements.txt +++ b/pip-requirements.txt @@ -1,5 +1,5 @@ pretrainedmodels==0.7.4 efficientnet-pytorch==0.5.1 albumentations==0.3.0 -torch==1.2.0 +torch==2.2.0 torchvision==0.4.0 \ No newline at end of file