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

Before Change

        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:])

    else:
        network_init_fxn = eval("innvestigate.utils.tests.networks.base.{}".format(modelname))

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)
        network = network_init_fxn(input_shape,
                                   num_classes,
                                   **kwargs)
        model_wo_sm = Model(inputs=network["in"], outputs=network["out"])

Instances