Update train.py

train.py

@@ -10,6 +10,7 @@ from datasets import load_dataset
 from transformers import AutoTokenizer
 
 tokenizer = AutoTokenizer.from_pretrained("bert-base-chinese", use_fast=True)
+import BertTokenizer  # 关键：新增这行导入
 
 import torch
 import numpy as np