Fixed typo

pkmr06 · Jul 31, 2017 · 6b7b087 · 6b7b087
1 parent 01b9411
commit 6b7b087
Show file tree

Hide file tree

Showing 8 changed files with 7 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -4,4 +4,4 @@
 .python-version
 __pycache__
 images
-vis.py
+results_
diff --git a/lib/gradients.py b/lib/gradients.py
@@ -40,6 +40,12 @@ class SmoothGrad(VanillaGrad):
     def __init__(self, pretrained_model, cuda=False, stdev_spread=0.15,
                  n_samples=25, magnitude=True):
         super(SmoothGrad, self).__init__(pretrained_model, cuda)
+        """
+        self.pretrained_model = pretrained_model
+        self.features = pretrained_model.features
+        self.cuda = cuda
+        self.pretrained_model.eval()
+        """
         self.stdev_spread = stdev_spread
         self.n_samples = n_samples
         self.magnitutde = magnitude

diff --git a/result/cam/grad_cam.jpg → results/cam/grad_cam.jpg b/result/cam/grad_cam.jpg → results/cam/grad_cam.jpg
diff --git a/result/cam/guided_grad_cam.jpg → results/cam/guided_grad_cam.jpg b/result/cam/guided_grad_cam.jpg → results/cam/guided_grad_cam.jpg
diff --git a/result/grad/guided_grad.jpg → results/grad/guided_grad.jpg b/result/grad/guided_grad.jpg → results/grad/guided_grad.jpg
diff --git a/result/grad/guided_smooth_grad.jpg → results/grad/guided_smooth_grad.jpg b/result/grad/guided_smooth_grad.jpg → results/grad/guided_smooth_grad.jpg
diff --git a/result/grad/smooth_grad.jpg → results/grad/smooth_grad.jpg b/result/grad/smooth_grad.jpg → results/grad/smooth_grad.jpg
diff --git a/result/grad/vanilla_grad.jpg → results/grad/vanilla_grad.jpg b/result/grad/vanilla_grad.jpg → results/grad/vanilla_grad.jpg