diff --git a/requirements.txt b/requirements.txt index 0660ee7..8fa93b9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,5 +13,5 @@ visdom bs4 html5lib ninja -torch==0.4.1 +torch==2.2.0 torchvision==0.2.1 \ No newline at end of file