Update main.py

main.py

@@ -69,7 +69,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('/app/bert-base-zh')
+        self.bert = BertModel.from_pretrained('/home/user/app/./bert-base-zh')
         # self.bert_config = self.bert.config
         self.out = nn.Sequential(
             # nn.Linear(768,256),
@@ -166,7 +166,7 @@ def model_init():
     acuda = torch.cuda.is_available() and not args.no_cuda
     device = 'cuda' if acuda else 'cpu'
     os.environ["CUDA_VISIBLE_DEVICES"] = device
-    tokenizer = BertTokenizerFast(vocab_file='/app/bert-base-zh/vocab.txt', sep_token="[SEP]", pad_token="[PAD]", cls_token="[CLS]")
+    tokenizer = BertTokenizerFast(vocab_file='/home/user/app/./bert-base-zh/vocab.txt', sep_token="[SEP]", pad_token="[PAD]", cls_token="[CLS]")
     # # tokenizer = BertTokenizer(vocab_file=args.voca_path)
     model = Word_BERT()
     # # model = model.load_state_dict(torch.load(args.model_path))