Test IntegratedGradients as well

Author: adrhill

src/input_augmentation.jl

@@ -123,9 +123,9 @@ function call_analyzer(
     # Further augmentations
     input_delta = (input - input_ref) / (aug.n - 1)
     for _ in 1:(aug.n)
-        input_aug += input_delta
+        input_aug .+= input_delta
         expl_aug = aug.analyzer(input_aug, output_selector)
-        sum_val += expl_aug.val
+        sum_val .+= expl_aug.val
     end
 
     # Average gradients and compute explanation

test/test_gpu.jl

@@ -20,7 +20,7 @@ model_gpu = device(model)
 input_gpu = device(input)
 @test_nowarn model_gpu(input_gpu)
 
-analyzer_types = (Gradient, SmoothGrad, InputTimesGradient)
+analyzer_types = (Gradient, SmoothGrad, InputTimesGradient, IntegratedGradients)
 
 @testset "Run analyzer (CPU)" begin
     @testset "$A" for A in analyzer_types