diff --git a/requirements.txt b/requirements.txt index beccb4b..0079535 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch ==1.6.0 +torch ==2.2.0 torchvision ==0.7.0 tensorboard numpy