errors about loading pre-train models

[harrywuhust2022]

KeyError: "Unable to open object (object 'branch1.0.conv.weight' doesn't exist)"

that's the error I have encounter when I write the code like this:

model_path = './mcnn_shtechA_660.h5'

model = MCNN()

trained_model = os.path.join(model_path)

load_net(trained_model, model)

model = model.to(device)