Updated notebook (#2909)

- fixed bug with Resize and size as single int

Updated notebook (#2909)
- fixed bug with Resize and size as single int
c6156080 · vfdev · GitHub · e8b6e3f0 · c6156080
Unverified Commit c6156080 authored Oct 27, 2020 by vfdev Committed by GitHub Oct 27, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

examples/python/tensor_transforms.ipynb examples/python/tensor_transforms.ipynb +2 -2

No files found.
--- a/examples/python/tensor_transforms.ipynb
+++ b/examples/python/tensor_transforms.ipynb
@@ -206,7 +206,7 @@
    "        super().__init__()\n",
    "        self.resnet18 = resnet18(pretrained=True).eval()\n",
    "        self.transforms = nn.Sequential(\n",
-    "            T.Resize(256),\n",
+    "            T.Resize([256, ]), # We use single int value inside a list due to torchscript type restrictions\n",
    "            T.CenterCrop(224),\n",
    "            T.ConvertImageDtype(torch.float),\n",
    "            T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])\n",
@@ -380,7 +380,7 @@
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
-   "version": "3.7.7"
+   "version": "3.7.4"
  }
 },
 "nbformat": 4,