Update main.py

main.py

@@ -38,7 +38,7 @@ def set_args():
     #                     help='模型参数')
     parser.add_argument('--log_path', default='interact.log', type=str, required=False, help='interact日志存放位置')
     parser.add_argument('--model_path', default='./pathology_extra/result/12/model.pth', type=str, required=False, help='对话模型路径')
-    parser.add_argument('--vocab_path', default='D:\\transformerFileDownload\\Pytorch\\bert-base-zh\\vocab.txt', type=str, required=False,
+    parser.add_argument('--vocab_path', default='./bert-base-zh\\vocab.txt', type=str, required=False,
                         help='对话模型路径')
     parser.add_argument('--save_samples_path', default="sample/", type=str, required=False, help="保存聊天记录的文件路径")
     parser.add_argument('--repetition_penalty', default=1.0, type=float, required=False,
@@ -78,7 +78,7 @@ def create_logger(args):
 class Word_BERT(nn.Module):
     def __init__(self, seq_label=1,cancer_label=8,transfer_label=2,ly_transfer=2):
         super(Word_BERT, self).__init__()
-        self.bert = BertModel.from_pretrained('D:\\transformerFileDownload\\Pytorch\\bert-base-zh')
+        self.bert = BertModel.from_pretrained('./bert-base-zh')
         # self.bert_config = self.bert.config
         self.out = nn.Sequential(
             # nn.Linear(768,256),