pycorrector icon indicating copy to clipboard operation
pycorrector copied to clipboard

macbert微调保存模型结构的问题

Open evanchenhi opened this issue 1 year ago • 2 comments

我看pycorrector/macbert/train_macbert4csc.yml配置文件里BERT_CKPT是原始 hfl/chinese-macbert-base 模型

微调完保存的也是原始模型,链接如下,并没有保存MacBert4Csc模型结构,只保存了权重参数 bert = BertForMaskedLM.from_pretrained(cfg.MODEL.BERT_CKPT) bert.save_pretrained(cfg.OUTPUT_DIR) https://github.com/shibing624/pycorrector/blob/master/pycorrector/macbert/train.py#L114

不知道我看的是否正确

evanchenhi avatar Oct 17 '23 08:10 evanchenhi

shibing624 avatar Oct 18 '23 03:10 shibing624

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.(由于长期不活动,机器人自动关闭此问题,如果需要欢迎提问)

stale[bot] avatar Dec 27 '23 07:12 stale[bot]