6c06a88957549b1405f43361e4aa911abc56ef66,examples/shrec17/test.py,,main,#Any#Any#Any#Any#Any#,24

Before Change

    model = mod.Model(55).to(device)

    model.load_state_dict(torch.load(os.path.join(log_dir, "state.pkl")))

    resdir = os.path.join(log_dir, dataset + "_perturbed")
    if os.path.isdir(resdir):
        shutil.rmtree(resdir)

After Change

    model = mod.Model(55).to(device)

    state = torch.load(os.path.join(log_dir, "state.pkl"))
    state = { key.replace("conv.kernel", "kernel.kernel").replace("conv.weight", "kernel.weight") : value for key, value in state.items() }
    model.load_state_dict(state)

    resdir = os.path.join(log_dir, dataset + "_perturbed")
    if os.path.isdir(resdir):
        shutil.rmtree(resdir)

Instances