jd-aig / nlp_baai

NLP models and codes for BAAI-JD joint project.
252 stars 56 forks source link

transformer加载预训练模型出现Segmentation fault (core dumped)错误。 #13

Closed QiuSYang closed 4 years ago

QiuSYang commented 4 years ago

model = model_class.from_pretrained(args.model_name_or_path, from_tf=bool('.ckpt' in args.model_name_or_path), config=config),在进行检索baseline model调试过程中出现预训练模型加载时出现Segmentation fault (core dumped)。 transformer version:2.9.1, 2.11.0 都会出现这样的错误 pytorch version: 1.3.1