ERNIE-text-classification-pytorch icon indicating copy to clipboard operation
ERNIE-text-classification-pytorch copied to clipboard

you may miss something the model needed

Open yuanjun1991 opened this issue 5 years ago • 1 comments

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'

yuanjun1991 avatar May 31 '19 14:05 yuanjun1991

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

lonePatient avatar May 31 '19 15:05 lonePatient