Skip to content

Commit

Permalink
fixed import error
Browse files Browse the repository at this point in the history
  • Loading branch information
@LostOxygen
LostOxygen committed Jun 27, 2023
1 parent 533dd92 commit 7fb8eec
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions visualize_lrp.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,9 @@

import torch
from torch.utils.data import DataLoader
from captum import LRP
from captum.attr import LRP
from captum.attr import visualization as viz
import numpy as np
# import numpy as np
from matplotlib import pyplot as plt

from kernel_eval.models import vgg11, vgg13, vgg16, vgg19, resnet34
Expand Down

0 comments on commit 7fb8eec

Please sign in to comment.