3033c545288896720b201183ed0d67d910710e47,examples/mnist_lrp.py,,create_model,#Any#Any#,70

Before Change

    if not modelname in innvestigate.utils.tests.networks.base.__all__:
        raise ValueError("Unknown model name {}".format(modelname))

    if modelname.startswith("pt_"): // load PreTrained models
        model_init_fxn = eval("innvestigate.utils.tests.networks.base.{}".format(modelname))
        model_wo_sm, model_w_sm = model_init_fxn(input_shape[1:])

After Change

    if modelname in innvestigate.applications.mnist.__all__: // load PreTrained models
        model_init_fxn = getattr(innvestigate.applications.mnist, modelname)
        model_wo_sm, model_w_sm = model_init_fxn(input_shape[1:])

    elif modelname in innvestigate.utils.tests.networks.base.__all__:
        network_init_fxn = getattr(innvestigate.utils.tests.networks.base, modelname)

Instances