今天调试代码的时候发现一个错误:
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)
今天调试代码的时候发现一个错误:
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)