diff --git a/docs/notebooks/transfer_learning/requirements.txt b/docs/notebooks/transfer_learning/requirements.txt index 2a953cf53..9be0d3dae 100644 --- a/docs/notebooks/transfer_learning/requirements.txt +++ b/docs/notebooks/transfer_learning/requirements.txt @@ -28,7 +28,7 @@ tensorflow-addons~=0.20.0 tensorflow-datasets~=4.9.2 tensorflow-hub~=0.13.0 tf-models-official -torch==1.13.1 +torch==2.2.0 torchvision==0.14.1 transformers~=4.38.0 # upgraded to avoid a vulnerability urllib3>=2.2.2