diff --git a/requirements.txt b/requirements.txt index f8f985b..d702e9c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -97,7 +97,7 @@ tensorboard-plugin-wit==1.8.1 tensorboardx==2.6 threadpoolctl==3.1.0 tomli==2.0.1 -torch==1.12.1 +torch==2.2.0 torchvision==0.13.1 tqdm==4.65.0 typed-argument-parser==1.7.2