Commit daeb6af6 authored by Ahmed Abdo's avatar Ahmed Abdo Committed by GitHub
Browse files

fixing densenet161 reference

parent d3e78103
...@@ -20,7 +20,7 @@ You can construct a model with random weights by calling its constructor: ...@@ -20,7 +20,7 @@ You can construct a model with random weights by calling its constructor:
alexnet = models.alexnet() alexnet = models.alexnet()
vgg16 = models.vgg16() vgg16 = models.vgg16()
squeezenet = models.squeezenet1_0() squeezenet = models.squeezenet1_0()
densenet = models.densenet_161() densenet = models.densenet161()
inception = models.inception_v3() inception = models.inception_v3()
We provide pre-trained models, using the PyTorch :mod:`torch.utils.model_zoo`. We provide pre-trained models, using the PyTorch :mod:`torch.utils.model_zoo`.
...@@ -33,7 +33,7 @@ These can be constructed by passing ``pretrained=True``: ...@@ -33,7 +33,7 @@ These can be constructed by passing ``pretrained=True``:
alexnet = models.alexnet(pretrained=True) alexnet = models.alexnet(pretrained=True)
squeezenet = models.squeezenet1_0(pretrained=True) squeezenet = models.squeezenet1_0(pretrained=True)
vgg16 = models.vgg16(pretrained=True) vgg16 = models.vgg16(pretrained=True)
densenet = models.densenet_161(pretrained=True) densenet = models.densenet161(pretrained=True)
inception = models.inception_v3(pretrained=True) inception = models.inception_v3(pretrained=True)
All pre-trained models expect input images normalized in the same way, All pre-trained models expect input images normalized in the same way,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment