diff --git a/lrp_utils.py b/lrp_utils.py index 1473d55..106829c 100644 --- a/lrp_utils.py +++ b/lrp_utils.py @@ -219,5 +219,4 @@ def __init__( (Convolution, WSquare(stabilizer=stabilizer, **rule_kwargs)) ] super().__init__(layer_map=layer_map, first_map=first_map, canonizers=canonizers) -from zennit.composites import EpsilonPlusFlat -composite_sample_cnn = EpsilonPlusWsquare(canonizers=[SampleCNNCanonizer()]) \ No newline at end of file +composite_sample_cnn = EpsilonPlusWsquare(canonizers=[SampleCNNCanonizer()])