能否把optimization和pytorch的checkpoint这个也放进来,我用最新的bert-pytorch master的代码转的checkpoint报错:
model.bert.load_state_dict(torch.load(args.init_checkpoint, map_location='cpu'))
RuntimeError: Error(s) in loading state_dict for BertModel:
Missing key(s) in state_dict: