Commit 88e81cea authored by Po-Hsien Chu's avatar Po-Hsien Chu Committed by Francisco Massa
Browse files

fix densenet example in README (#280)

parent 87fe75b0
......@@ -266,7 +266,7 @@ constructor:
alexnet = models.alexnet()
vgg16 = models.vgg16()
squeezenet = models.squeezenet1_0()
densenet = models.densenet_161()
densenet = models.densenet161()
inception = models.inception_v3()
We provide pre-trained models for the ResNet variants, SqueezeNet 1.0 and 1.1,
......@@ -280,7 +280,7 @@ These can be constructed by passing ``pretrained=True``:
alexnet = models.alexnet(pretrained=True)
squeezenet = models.squeezenet1_0(pretrained=True)
vgg16 = models.vgg16(pretrained=True)
densenet = models.densenet_161(pretrained=True)
densenet = models.densenet161(pretrained=True)
inception = models.inception_v3(pretrained=True)
......
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