bwedin

One-line change: adds output_layer = output_layer[:,target_class_idx] to the else statement in def call_model_function. This means that previously Grad-CAM was calculating wrt the entire output layer, this should fix it to...