Missing key(s) in state_dict:

今天调试代码的时候发现一个错误:
Missing key(s) in state_dict:

错误代码:

model.bert.load_state_dict(torch.load(args.init_checkpoint, map_location="cpu"))

解决方案:

model.load_state_dict(torch.load(args.init_checkpoint), False)

你可能感兴趣的:(解决的bug,python)