you may miss something the model needed

[yuanjun1991]

when i download the ernie_base from panbaidu,and run fine_tune_ernie.py , it raise error as follows:

File "/Users/yuanjun/Downloads/ERNIE-text-classification-pytorch-master/fine_tune_ernie.py", line 8, in
from pyernie.io.dataset import CreateDataset
File "/Users/yuanjun/Downloads/ERNIE-text-classification-pytorch-master/pyernie/io/dataset.py", line 5, in
from pytorch_pretrained_bert.tokenization import BertTokenizer
ModuleNotFoundError: No module named 'pytorch_pretrained_bert'

[lonePatient]

@yuanjun1991 ｈｉ,in file "/pyernie/io/dataset.py", line 5 , changed to from ..model.ernie.tokenization import BertTokenizer