能否把optimization和pytorch的checkpoint这个也放进来

[ruoji]

能否把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:

[NLPScott]

可以看下这个https://github.com/NLPScott/pytorch-pretrained-BERT