Update IMDB_TorchText_Interpret.ipynb (#764)

bilalsal · facebook-github-bot · commit d502e05992ef · 2021-09-15T09:43:26.000-07:00
Summary: Fix #763 Pull Request resolved: #764 Reviewed By: vivekmig Differential Revision: D30961719 Pulled By: bilalsal fbshipit-source-id: a1840dc092486eed6ce711ebb307fe78c21d6999
diff --git a/tutorials/IMDB_TorchText_Interpret.ipynb b/tutorials/IMDB_TorchText_Interpret.ipynb
@@ -148,7 +148,7 @@
    "source": [
     "Loads pretrained model and sets the model to eval mode.\n",
     "\n",
-    "The model can be downloaded here: https://github.com/pytorch/captum/blob/master/tutorials/models/imdb-model-cnn.pt"
+    "The model can be downloaded here: https://github.com/pytorch/captum/blob/master/tutorials/models/imdb-model-cnn-large.pt"
    ]
   },
   {
@@ -363,7 +363,7 @@
     "                            Label.vocab.itos[pred_ind],\n",
     "                            Label.vocab.itos[label],\n",
     "                            Label.vocab.itos[1],\n",
-    "                            attributions.sum(),       \n",
+    "                            attributions.sum(),\n",
     "                            text,\n",
     "                            delta))"
    ]

Original file line number	Diff line number	Diff line change
`@@ -148,7 +148,7 @@`
`148`	`148`	`"source": [`
`149`	`149`	`"Loads pretrained model and sets the model to eval mode.\n",`
`150`	`150`	`"\n",`
`151`		`- "The model can be downloaded here: https://github.com/pytorch/captum/blob/master/tutorials/models/imdb-model-cnn.pt"`
	`151`	`+ "The model can be downloaded here: https://github.com/pytorch/captum/blob/master/tutorials/models/imdb-model-cnn-large.pt"`
`152`	`152`	`]`
`153`	`153`	`},`
`154`	`154`	`{`
`@@ -363,7 +363,7 @@`
`363`	`363`	`" Label.vocab.itos[pred_ind],\n",`
`364`	`364`	`" Label.vocab.itos[label],\n",`
`365`	`365`	`" Label.vocab.itos[1],\n",`
`366`		`- " attributions.sum(), \n",`
	`366`	`+ " attributions.sum(),\n",`
`367`	`367`	`" text,\n",`
`368`	`368`	`" delta))"`
`369`	`369`	`]`