diff --git a/requirements.txt b/requirements.txt index f3f355a..cdec1f4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ scipy==1.1.0 horovod==0.16.0 -torch==1.5.0 +torch==1.13.1 torchvision==0.6.0 six==1.11.0 imageio==2.8.0