From dbf6c3d7e88e789e772cd24a650ecddf1f94926c Mon Sep 17 00:00:00 2001 From: Ahmed Nabil <72295771+AI-Ahmed@users.noreply.github.com> Date: Tue, 13 Dec 2022 16:16:00 +0200 Subject: [PATCH 1/2] Add the image's type to the `image` parameter. --- week07_interpretability/practice.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/week07_interpretability/practice.ipynb b/week07_interpretability/practice.ipynb index d2efd4083..1018ae073 100644 --- a/week07_interpretability/practice.ipynb +++ b/week07_interpretability/practice.ipynb @@ -150,7 +150,7 @@ "metadata": {}, "outputs": [], "source": [ - "def explain_smoothgrad(image: torch., class_ix, num_runs=32, sigma=0.1):\n", + "def explain_smoothgrad(image: torch.Tensor, class_ix, num_runs=32, sigma=0.1):\n", " image_tensor = torch.as_tensor(image, device=device, dtype=torch.float32)\n", " image_tensor = image_tensor[None].permute(0, 3, 1, 2) # shape: [1, 3, h, w]\n", " \n", From 9f57d72819ddd541854bd1308c0da36ceee0944a Mon Sep 17 00:00:00 2001 From: Ahmed Nabil <72295771+AI-Ahmed@users.noreply.github.com> Date: Tue, 13 Dec 2022 16:18:38 +0200 Subject: [PATCH 2/2] Update practice.ipynb --- week07_interpretability/practice.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/week07_interpretability/practice.ipynb b/week07_interpretability/practice.ipynb index 1018ae073..63c95319b 100644 --- a/week07_interpretability/practice.ipynb +++ b/week07_interpretability/practice.ipynb @@ -150,7 +150,7 @@ "metadata": {}, "outputs": [], "source": [ - "def explain_smoothgrad(image: torch.Tensor, class_ix, num_runs=32, sigma=0.1):\n", + "def explain_smoothgrad(image: np.ndarray, class_ix, num_runs=32, sigma=0.1):\n", " image_tensor = torch.as_tensor(image, device=device, dtype=torch.float32)\n", " image_tensor = image_tensor[None].permute(0, 3, 1, 2) # shape: [1, 3, h, w]\n", " \n",