itecture
    The reason we can&#39;t automatically get this score with cd_generic is because
    the model.modules() is missing some things like self.maxpool, and self.Relu
    because the model file only defined these things in the forward method
    &#39;&#39;&#39;
    mods = list(model.modules())[1:]
    relevant, irrelevant = propagate_conv_linear(relevant, irrelevant, mods[0])
    relevant, irrelevant = propagate_pooling(relevant, irrelevant,
                                             lambda x: F.max