Hi I met the same problem but
after I changed the code like this:
state_dict =checkpoint['model_dict']
state_dict = {k.partition('model.')[2]: v for k,v in state_dict}
model.load_state_dict(state_dict)

it showed this error:
ValueError: too many values to unpack (expected 2)
why it didn’t work?