p71924506
/
deepke
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

fix bug

This commit is contained in:
Xin Xu 2021-11-13 17:30:22 +08:00 committed by GitHub
parent dc84f31e40
commit ca21d5148a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
example/ner/standard/run.py
View File

@ -159,7 +159,7 @@ def main(cfg):
# Load a trained model and config that you have fine-tuned
else:
# Load a trained model and vocabulary that you have fine-tuned
model = Ner.from_pretrained(utils.get_original_cwd()+'/'+cfg.output_dir)
model = TrainNer.from_pretrained(utils.get_original_cwd()+'/'+cfg.output_dir)
tokenizer = BertTokenizer.from_pretrained(utils.get_original_cwd()+'/'+cfg.output_dir, do_lower_case=cfg.do_lower_case)
model.to(device)
@ -227,4 +227,4 @@ def main(cfg):
writer.write(report)
if __name__ == '__main__':
main()
main()