Merge pull request #65 from jspisak/patch-4

ericnakagawa · web-flow · commit f0665f09c43f · 2018-09-18T14:04:35.000-07:00
Update tools-libraries.md
diff --git a/_features/tools-libraries.md b/_features/tools-libraries.md
@@ -3,24 +3,13 @@ title: Tools & Libraries
 order: 4
 snippet: >
   ```python
-    from torchvision import transforms, utils
-    
-    scale = Rescale(256)
-    crop = RandomCrop(128)
-    composed = transforms.Compose([Rescale(256),
-    RandomCrop(224)])
-
-    fig = plt.figure()
-    sample = face_dataset[65]
-    for i, tsfrm in enumerate([scale, crop, composed]):
-    transformed_sample = tsfrm(sample)
-
-    ax = plt.subplot(1, 3, i + 1)
-    plt.tight_layout()
-    ax.set_title(type(tsfrm).__name__)
-    show_landmarks(**transformed_sample)
-
-    plt.show()
+    import torchvision.models as models
+    resnet18 = models.resnet18(pretrained=True)
+    alexnet = models.alexnet(pretrained=True)
+    squeezenet = models.squeezenet1_0(pretrained=True)
+    vgg16 = models.vgg16(pretrained=True)
+    densenet = models.densenet161(pretrained=True)
+    inception = models.inception_v3(pretrained=True)
   ```
 
 summary-home: A rich ecosystem of tools and libraries extends PyTorch and supports development in computer vision, NLP and more.
