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..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.10.0', 'alexnet', pretrained=True) +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 ecc52607..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.10.0', 'deeplabv3_resnet50', pretrained=True) +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.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.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 29cab0a8..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.10.0', 'densenet121', pretrained=True) +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.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.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 bb501f26..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.10.0', 'fcn_resnet50', pretrained=True) +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.10.0', 'fcn_resnet101', pretrained=True) +# 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 0e8f1d25..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.10.0', 'googlenet', pretrained=True) +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 a87e4d86..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.10.0', 'inception_v3', pretrained=True) +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 6be178b5..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.10.0', 'mobilenet_v2', pretrained=True) +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 a64244a1..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.10.0', 'resnet18', pretrained=True) +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.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.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 055c1abc..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.10.0', 'resnext50_32x4d', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.14.1', '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.14.1', 'resnext101_32x8d', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_shufflenet_v2.md b/pytorch_vision_shufflenet_v2.md index 1fac4538..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.10.0', 'shufflenet_v2_x1_0', pretrained=True) +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 bff65a8c..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.10.0', 'squeezenet1_0', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.14.1', '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.14.1', 'squeezenet1_1', weights="IMAGENET1K_V1") model.eval() ``` diff --git a/pytorch_vision_vgg.md b/pytorch_vision_vgg.md index 4e59ab22..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.10.0', 'vgg11', pretrained=True) +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.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.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 2419a807..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.10.0', 'wide_resnet50_2', pretrained=True) +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.10.0', 'wide_resnet101_2', pretrained=True) +model = torch.hub.load('pytorch/vision:v0.14.1', 'wide_resnet101_2', weights="IMAGENET1K_V1") model.eval() ```