pytorch LRP的实现
PyTorch对线性层和卷积层的一些层明智相关性传播(LRP)规则[1、2、3]的实现。
这些模块装饰了torch.nn.Sequential , torch.nn.Linear和torch.nn.Conv2d ,以便能够使用autograd算法来计算说明。
安装
要安装需求,请参阅文件。
如果使用conda ,则可以通过执行以下命令来安装称为torchlrp的环境:
> conda env create -f requirements.yml
为了能够如下所示导入lrp ,请确保路径中包含TorchLRP目录。
用法
该代码可以按如下方式使用:
import torch
import lrp
model = Sequential (
lrp . Conv2d ( 1 , 32 , 3 , 1 , 1 ),
torch . nn
1