From 48ff225849c35d9581750aea8877924c577b1d21 Mon Sep 17 00:00:00 2001 From: Bilal Alsallakh Date: Tue, 8 Dec 2020 11:04:32 -0800 Subject: [PATCH] Remove trailing comma The comma is problematic as reported in https://github.com/pytorch/captum/issues/553 --- recipes_source/recipes/Captum_Recipe.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/recipes_source/recipes/Captum_Recipe.py b/recipes_source/recipes/Captum_Recipe.py index 105c8cd5f67..e0a99796907 100644 --- a/recipes_source/recipes/Captum_Recipe.py +++ b/recipes_source/recipes/Captum_Recipe.py @@ -136,7 +136,7 @@ attribution_dog = np.transpose(attribution_dog.squeeze().cpu().detach().numpy(), (1,2,0)) vis_types = ["heat_map", "original_image"] -vis_signs = ["all", "all"], # "positive", "negative", or "all" to show both +vis_signs = ["all", "all"] # "positive", "negative", or "all" to show both # positive attribution indicates that the presence of the area increases the prediction score # negative attribution indicates distractor areas whose absence increases the score @@ -186,4 +186,4 @@ # # Another useful post by Gilbert Tanner: # https://gilberttanner.com/blog/interpreting-pytorch-models-with-captum -# \ No newline at end of file +#