diff --git a/model_templates/python3_pytorch_multiclass/requirements.txt b/model_templates/python3_pytorch_multiclass/requirements.txt index 34cd62370..b07055eb1 100644 --- a/model_templates/python3_pytorch_multiclass/requirements.txt +++ b/model_templates/python3_pytorch_multiclass/requirements.txt @@ -1,3 +1,3 @@ -torch==1.13.1 +torch==2.7.1 scikit-learn==1.0.2 scipy>=1.1,<1.11