bert-Chinese-classification-task icon indicating copy to clipboard operation
bert-Chinese-classification-task copied to clipboard

Published 20 hours ago • verified publisher icon NLPScott

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

Open ruoji opened this issue 6 years ago • 1 comments

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

ruoji avatar Dec 03 '18 10:12 ruoji

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

NLPScott avatar Dec 11 '18 07:12 NLPScott