diff --git a/scripts/requirements.txt b/scripts/requirements.txt index 406a2b08..417dd2bf 100644 --- a/scripts/requirements.txt +++ b/scripts/requirements.txt @@ -1,4 +1,4 @@ -torch>=1.10.2 +torch>=2.0.0 torchvision>=0.11.3 pytorch-ignite>=0.4.8 pyyaml diff --git a/src/templates/template-common/requirements.txt b/src/templates/template-common/requirements.txt index 1f9dbbdb..b7bb4ead 100644 --- a/src/templates/template-common/requirements.txt +++ b/src/templates/template-common/requirements.txt @@ -1,4 +1,4 @@ -torch>=1.10.2 +torch>=2.0.0 torchvision>=0.11.3 pytorch-ignite>=0.4.8 pyyaml diff --git a/src/templates/template-text-classification/requirements.txt b/src/templates/template-text-classification/requirements.txt index 0bbbc09b..04d1a879 100644 --- a/src/templates/template-text-classification/requirements.txt +++ b/src/templates/template-text-classification/requirements.txt @@ -1,4 +1,4 @@ -torch>=1.10.2 +torch>=2.0.0 torchvision>=0.11.3 pytorch-ignite>=0.4.8 pyyaml diff --git a/src/templates/template-vision-classification/requirements.txt b/src/templates/template-vision-classification/requirements.txt index 1f9dbbdb..b7bb4ead 100644 --- a/src/templates/template-vision-classification/requirements.txt +++ b/src/templates/template-vision-classification/requirements.txt @@ -1,4 +1,4 @@ -torch>=1.10.2 +torch>=2.0.0 torchvision>=0.11.3 pytorch-ignite>=0.4.8 pyyaml diff --git a/src/templates/template-vision-dcgan/requirements.txt b/src/templates/template-vision-dcgan/requirements.txt index 1f9dbbdb..b7bb4ead 100644 --- a/src/templates/template-vision-dcgan/requirements.txt +++ b/src/templates/template-vision-dcgan/requirements.txt @@ -1,4 +1,4 @@ -torch>=1.10.2 +torch>=2.0.0 torchvision>=0.11.3 pytorch-ignite>=0.4.8 pyyaml diff --git a/src/templates/template-vision-segmentation/requirements.txt b/src/templates/template-vision-segmentation/requirements.txt index 40373974..d5ae9af8 100644 --- a/src/templates/template-vision-segmentation/requirements.txt +++ b/src/templates/template-vision-segmentation/requirements.txt @@ -1,4 +1,4 @@ -torch>=1.10.2 +torch>=2.0.0 torchvision>=0.11.3 pytorch-ignite>=0.4.8 pyyaml