JasonLiao commited on
Commit
31545d7
1 Parent(s): 7737d00

fix the method load the pretrained model

Browse files
Files changed (1) hide show
  1. code/args.py +1 -1
code/args.py CHANGED
@@ -4,7 +4,7 @@ class args():
4
 
5
  #pre_model_name = "bert-base-chinese"
6
  #pre_model_name = "hfl/chinese-macbert-base"
7
- pre_model_name = "hfl/chinese-roberta-wwm-ext"
8
  save_model_name = "roberta_crf"
9
 
10
  LOG_DIR = "../log/long_term/"+save_model_name+"/"
 
4
 
5
  #pre_model_name = "bert-base-chinese"
6
  #pre_model_name = "hfl/chinese-macbert-base"
7
+ pre_model_name = "chinese-roberta-wwm-ext"
8
  save_model_name = "roberta_crf"
9
 
10
  LOG_DIR = "../log/long_term/"+save_model_name+"/"