From e6cd1555e6b56bba3aa477017be79023cbffad0a Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Fri, 15 Jul 2022 16:41:15 +0100 Subject: [PATCH 1/2] Update md file to torchvision 0.13 and use new MultiWeights API --- docs/template.md | 2 +- pytorch_vision_alexnet.md | 2 +- pytorch_vision_deeplabv3_resnet101.md | 6 +++--- pytorch_vision_densenet.md | 8 ++++---- pytorch_vision_fcn_resnet101.md | 4 ++-- pytorch_vision_googlenet.md | 2 +- pytorch_vision_inception_v3.md | 2 +- pytorch_vision_mobilenet_v2.md | 2 +- pytorch_vision_resnet.md | 10 +++++----- pytorch_vision_resnext.md | 4 ++-- pytorch_vision_shufflenet_v2.md | 2 +- pytorch_vision_squeezenet.md | 4 ++-- pytorch_vision_vgg.md | 16 ++++++++-------- pytorch_vision_wide_resnet.md | 4 ++-- 14 files changed, 34 insertions(+), 34 deletions(-) diff --git a/docs/template.md b/docs/template.md index be3b7a3e..a2b5e7b4 100644 --- a/docs/template.md +++ b/docs/template.md @@ -18,7 +18,7 @@ accelerator: ```python import torch -torch.hub.load('pytorch/vision', 'resnet18', pretrained=True) +torch.hub.load('pytorch/vision', 'resnet18', weights="DEFAULT") ``` diff --git a/pytorch_vision_alexnet.md b/pytorch_vision_alexnet.md index 3fa38f53..3e70d0d5 100644 --- a/pytorch_vision_alexnet.md +++ b/pytorch_vision_alexnet.md @@ -19,7 +19,7 @@ demo-model-link: https://huggingface.co/spaces/pytorch/AlexNet ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'alexnet', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'alexnet', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_deeplabv3_resnet101.md b/pytorch_vision_deeplabv3_resnet101.md index ecc52607..ec5efb75 100644 --- a/pytorch_vision_deeplabv3_resnet101.md +++ b/pytorch_vision_deeplabv3_resnet101.md @@ -19,10 +19,10 @@ order: 1 ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'deeplabv3_resnet50', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'deeplabv3_resnet50', weights="COCO_WITH_VOC_LABELS_V1") # or any of these variants -# model = torch.hub.load('pytorch/vision:v0.10.0', 'deeplabv3_resnet101', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'deeplabv3_mobilenet_v3_large', pretrained=True) +# model = torch.hub.load('pytorch/vision:v0.13.0', 'deeplabv3_resnet101', weights="COCO_WITH_VOC_LABELS_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'deeplabv3_mobilenet_v3_large', weights="COCO_WITH_VOC_LABELS_V1") model.eval() ``` diff --git a/pytorch_vision_densenet.md b/pytorch_vision_densenet.md index 29cab0a8..594c8094 100644 --- a/pytorch_vision_densenet.md +++ b/pytorch_vision_densenet.md @@ -19,11 +19,11 @@ demo-model-link: https://huggingface.co/spaces/pytorch/Densenet ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'densenet121', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'densenet121', weights="IMAGENET1K_V1") # or any of these variants -# model = torch.hub.load('pytorch/vision:v0.10.0', 'densenet169', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'densenet201', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'densenet161', pretrained=True) +# model = torch.hub.load('pytorch/vision:v0.13.0', 'densenet169', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'densenet201', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'densenet161', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_fcn_resnet101.md b/pytorch_vision_fcn_resnet101.md index bb501f26..1441809e 100644 --- a/pytorch_vision_fcn_resnet101.md +++ b/pytorch_vision_fcn_resnet101.md @@ -19,9 +19,9 @@ demo-model-link: https://huggingface.co/spaces/pytorch/FCN ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'fcn_resnet50', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'fcn_resnet50', weights="COCO_WITH_VOC_LABELS_V1") # or -# model = torch.hub.load('pytorch/vision:v0.10.0', 'fcn_resnet101', pretrained=True) +# model = torch.hub.load('pytorch/vision:v0.13.0', 'fcn_resnet101', weights="COCO_WITH_VOC_LABELS_V1") model.eval() ``` diff --git a/pytorch_vision_googlenet.md b/pytorch_vision_googlenet.md index 0e8f1d25..9708ee10 100644 --- a/pytorch_vision_googlenet.md +++ b/pytorch_vision_googlenet.md @@ -19,7 +19,7 @@ order: 10 ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'googlenet', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'googlenet', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_inception_v3.md b/pytorch_vision_inception_v3.md index a87e4d86..09111162 100644 --- a/pytorch_vision_inception_v3.md +++ b/pytorch_vision_inception_v3.md @@ -19,7 +19,7 @@ demo-model-link: https://huggingface.co/spaces/pytorch/Inception_v3 ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'inception_v3', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'inception_v3', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_mobilenet_v2.md b/pytorch_vision_mobilenet_v2.md index 6be178b5..a85615d2 100644 --- a/pytorch_vision_mobilenet_v2.md +++ b/pytorch_vision_mobilenet_v2.md @@ -19,7 +19,7 @@ demo-model-link: https://huggingface.co/spaces/pytorch/MobileNet_v2 ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'mobilenet_v2', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'mobilenet_v2', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_resnet.md b/pytorch_vision_resnet.md index a64244a1..00d42a8a 100644 --- a/pytorch_vision_resnet.md +++ b/pytorch_vision_resnet.md @@ -19,12 +19,12 @@ demo-model-link: https://huggingface.co/spaces/pytorch/ResNet ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet18', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet18', weights="IMAGENET1K_V1") # or any of these variants -# model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet34', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet50', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet101', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet152', pretrained=True) +# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet34', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet50', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet101', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet152', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_resnext.md b/pytorch_vision_resnext.md index 055c1abc..8d53466c 100644 --- a/pytorch_vision_resnext.md +++ b/pytorch_vision_resnext.md @@ -19,9 +19,9 @@ demo-model-link: https://huggingface.co/spaces/pytorch/ResNext ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'resnext50_32x4d', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'resnext50_32x4d', weights="IMAGENET1K_V1") # or -# model = torch.hub.load('pytorch/vision:v0.10.0', 'resnext101_32x8d', pretrained=True) +# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnext101_32x8d', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_shufflenet_v2.md b/pytorch_vision_shufflenet_v2.md index 1fac4538..1f03affd 100644 --- a/pytorch_vision_shufflenet_v2.md +++ b/pytorch_vision_shufflenet_v2.md @@ -19,7 +19,7 @@ demo-model-link: https://huggingface.co/spaces/pytorch/ShuffleNet_v2 ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'shufflenet_v2_x1_0', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'shufflenet_v2_x1_0', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_squeezenet.md b/pytorch_vision_squeezenet.md index bff65a8c..2f38eafe 100644 --- a/pytorch_vision_squeezenet.md +++ b/pytorch_vision_squeezenet.md @@ -19,9 +19,9 @@ demo-model-link: https://huggingface.co/spaces/pytorch/SqueezeNet ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'squeezenet1_0', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'squeezenet1_0', weights="IMAGENET1K_V1") # or -# model = torch.hub.load('pytorch/vision:v0.10.0', 'squeezenet1_1', pretrained=True) +# model = torch.hub.load('pytorch/vision:v0.13.0', 'squeezenet1_1', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_vgg.md b/pytorch_vision_vgg.md index 4e59ab22..f94fa415 100644 --- a/pytorch_vision_vgg.md +++ b/pytorch_vision_vgg.md @@ -19,15 +19,15 @@ demo-model-link: https://huggingface.co/spaces/pytorch/vgg-nets ```python import torch -model = torch.hub.load('pytorch/vision:v0.10.0', 'vgg11', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg11', weights="IMAGENET1K_V1") # or any of these variants -# model = torch.hub.load('pytorch/vision:v0.10.0', 'vgg11_bn', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'vgg13', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'vgg13_bn', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'vgg16', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'vgg16_bn', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'vgg19', pretrained=True) -# model = torch.hub.load('pytorch/vision:v0.10.0', 'vgg19_bn', pretrained=True) +# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg11_bn', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg13', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg13_bn', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg16', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg16_bn', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg19', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg19_bn', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_wide_resnet.md b/pytorch_vision_wide_resnet.md index 2419a807..da58bdce 100644 --- a/pytorch_vision_wide_resnet.md +++ b/pytorch_vision_wide_resnet.md @@ -20,9 +20,9 @@ demo-model-link: https://huggingface.co/spaces/pytorch/Wide_Resnet ```python import torch # load WRN-50-2: -model = torch.hub.load('pytorch/vision:v0.10.0', 'wide_resnet50_2', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'wide_resnet50_2', weights="IMAGENET1K_V1") # or WRN-101-2 -model = torch.hub.load('pytorch/vision:v0.10.0', 'wide_resnet101_2', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.13.0', 'wide_resnet101_2', weights="IMAGENET1K_V1") model.eval() ``` From db634d807504ce929092d2291f3bdc59c3c50ce7 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Fri, 13 Jan 2023 11:21:09 +0000 Subject: [PATCH 2/2] 0.13 -> 0.14.1 --- pytorch_vision_alexnet.md | 2 +- pytorch_vision_deeplabv3_resnet101.md | 6 +++--- pytorch_vision_densenet.md | 8 ++++---- pytorch_vision_fcn_resnet101.md | 4 ++-- pytorch_vision_googlenet.md | 2 +- pytorch_vision_inception_v3.md | 2 +- pytorch_vision_mobilenet_v2.md | 2 +- pytorch_vision_resnet.md | 10 +++++----- pytorch_vision_resnext.md | 4 ++-- pytorch_vision_shufflenet_v2.md | 2 +- pytorch_vision_squeezenet.md | 4 ++-- pytorch_vision_vgg.md | 16 ++++++++-------- pytorch_vision_wide_resnet.md | 4 ++-- 13 files changed, 33 insertions(+), 33 deletions(-) diff --git a/pytorch_vision_alexnet.md b/pytorch_vision_alexnet.md index 3e70d0d5..39dc3625 100644 --- a/pytorch_vision_alexnet.md +++ b/pytorch_vision_alexnet.md @@ -19,7 +19,7 @@ demo-model-link: https://huggingface.co/spaces/pytorch/AlexNet ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'alexnet', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'alexnet', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_deeplabv3_resnet101.md b/pytorch_vision_deeplabv3_resnet101.md index ec5efb75..64531284 100644 --- a/pytorch_vision_deeplabv3_resnet101.md +++ b/pytorch_vision_deeplabv3_resnet101.md @@ -19,10 +19,10 @@ order: 1 ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'deeplabv3_resnet50', weights="COCO_WITH_VOC_LABELS_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'deeplabv3_resnet50', weights="COCO_WITH_VOC_LABELS_V1") # or any of these variants -# model = torch.hub.load('pytorch/vision:v0.13.0', 'deeplabv3_resnet101', weights="COCO_WITH_VOC_LABELS_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'deeplabv3_mobilenet_v3_large', weights="COCO_WITH_VOC_LABELS_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'deeplabv3_resnet101', weights="COCO_WITH_VOC_LABELS_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'deeplabv3_mobilenet_v3_large', weights="COCO_WITH_VOC_LABELS_V1") model.eval() ``` diff --git a/pytorch_vision_densenet.md b/pytorch_vision_densenet.md index 594c8094..edba7ccb 100644 --- a/pytorch_vision_densenet.md +++ b/pytorch_vision_densenet.md @@ -19,11 +19,11 @@ demo-model-link: https://huggingface.co/spaces/pytorch/Densenet ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'densenet121', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'densenet121', weights="IMAGENET1K_V1") # or any of these variants -# model = torch.hub.load('pytorch/vision:v0.13.0', 'densenet169', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'densenet201', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'densenet161', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'densenet169', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'densenet201', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'densenet161', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_fcn_resnet101.md b/pytorch_vision_fcn_resnet101.md index 1441809e..c7bc742f 100644 --- a/pytorch_vision_fcn_resnet101.md +++ b/pytorch_vision_fcn_resnet101.md @@ -19,9 +19,9 @@ demo-model-link: https://huggingface.co/spaces/pytorch/FCN ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'fcn_resnet50', weights="COCO_WITH_VOC_LABELS_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'fcn_resnet50', weights="COCO_WITH_VOC_LABELS_V1") # or -# model = torch.hub.load('pytorch/vision:v0.13.0', 'fcn_resnet101', weights="COCO_WITH_VOC_LABELS_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'fcn_resnet101', weights="COCO_WITH_VOC_LABELS_V1") model.eval() ``` diff --git a/pytorch_vision_googlenet.md b/pytorch_vision_googlenet.md index 9708ee10..1e85487a 100644 --- a/pytorch_vision_googlenet.md +++ b/pytorch_vision_googlenet.md @@ -19,7 +19,7 @@ order: 10 ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'googlenet', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'googlenet', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_inception_v3.md b/pytorch_vision_inception_v3.md index 09111162..449469d5 100644 --- a/pytorch_vision_inception_v3.md +++ b/pytorch_vision_inception_v3.md @@ -19,7 +19,7 @@ demo-model-link: https://huggingface.co/spaces/pytorch/Inception_v3 ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'inception_v3', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'inception_v3', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_mobilenet_v2.md b/pytorch_vision_mobilenet_v2.md index a85615d2..68f272ef 100644 --- a/pytorch_vision_mobilenet_v2.md +++ b/pytorch_vision_mobilenet_v2.md @@ -19,7 +19,7 @@ demo-model-link: https://huggingface.co/spaces/pytorch/MobileNet_v2 ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'mobilenet_v2', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'mobilenet_v2', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_resnet.md b/pytorch_vision_resnet.md index 00d42a8a..d54dfdc6 100644 --- a/pytorch_vision_resnet.md +++ b/pytorch_vision_resnet.md @@ -19,12 +19,12 @@ demo-model-link: https://huggingface.co/spaces/pytorch/ResNet ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet18', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'resnet18', weights="IMAGENET1K_V1") # or any of these variants -# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet34', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet50', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet101', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnet152', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'resnet34', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'resnet50', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'resnet101', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'resnet152', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_resnext.md b/pytorch_vision_resnext.md index 8d53466c..23512935 100644 --- a/pytorch_vision_resnext.md +++ b/pytorch_vision_resnext.md @@ -19,9 +19,9 @@ demo-model-link: https://huggingface.co/spaces/pytorch/ResNext ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'resnext50_32x4d', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'resnext50_32x4d', weights="IMAGENET1K_V1") # or -# model = torch.hub.load('pytorch/vision:v0.13.0', 'resnext101_32x8d', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'resnext101_32x8d', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_shufflenet_v2.md b/pytorch_vision_shufflenet_v2.md index 1f03affd..40c35b7a 100644 --- a/pytorch_vision_shufflenet_v2.md +++ b/pytorch_vision_shufflenet_v2.md @@ -19,7 +19,7 @@ demo-model-link: https://huggingface.co/spaces/pytorch/ShuffleNet_v2 ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'shufflenet_v2_x1_0', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'shufflenet_v2_x1_0', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_squeezenet.md b/pytorch_vision_squeezenet.md index 2f38eafe..2bbbb1c4 100644 --- a/pytorch_vision_squeezenet.md +++ b/pytorch_vision_squeezenet.md @@ -19,9 +19,9 @@ demo-model-link: https://huggingface.co/spaces/pytorch/SqueezeNet ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'squeezenet1_0', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'squeezenet1_0', weights="IMAGENET1K_V1") # or -# model = torch.hub.load('pytorch/vision:v0.13.0', 'squeezenet1_1', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'squeezenet1_1', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_vgg.md b/pytorch_vision_vgg.md index f94fa415..a375d02c 100644 --- a/pytorch_vision_vgg.md +++ b/pytorch_vision_vgg.md @@ -19,15 +19,15 @@ demo-model-link: https://huggingface.co/spaces/pytorch/vgg-nets ```python import torch -model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg11', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'vgg11', weights="IMAGENET1K_V1") # or any of these variants -# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg11_bn', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg13', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg13_bn', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg16', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg16_bn', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg19', weights="IMAGENET1K_V1") -# model = torch.hub.load('pytorch/vision:v0.13.0', 'vgg19_bn', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'vgg11_bn', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'vgg13', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'vgg13_bn', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'vgg16', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'vgg16_bn', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'vgg19', weights="IMAGENET1K_V1") +# model = torch.hub.load('pytorch/vision:v0.14.1', 'vgg19_bn', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_wide_resnet.md b/pytorch_vision_wide_resnet.md index da58bdce..94cb564f 100644 --- a/pytorch_vision_wide_resnet.md +++ b/pytorch_vision_wide_resnet.md @@ -20,9 +20,9 @@ demo-model-link: https://huggingface.co/spaces/pytorch/Wide_Resnet ```python import torch # load WRN-50-2: -model = torch.hub.load('pytorch/vision:v0.13.0', 'wide_resnet50_2', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'wide_resnet50_2', weights="IMAGENET1K_V1") # or WRN-101-2 -model = torch.hub.load('pytorch/vision:v0.13.0', 'wide_resnet101_2', weights="IMAGENET1K_V1") +model = torch.hub.load('pytorch/vision:v0.14.1', 'wide_resnet101_2', weights="IMAGENET1K_V1") model.eval() ```