search

liuyukid / transformers-ner

Pytorch-Named-Entity-Recognition-with-transformers
209 stars 44 forks source link

Issue of the path #11

Open blink7-lab opened 3 years ago

blink7-lab commented 3 years ago

你好请问下--model_type和--model_name_or_path的对应路径是什么呢？（我修改后一直报错OSERROR:can't load.....，特来向您请教。）

liuyukid commented 3 years ago

如果你用的是英文的BERT的话，可以这么写： MODEL_TYPE='bert'
MODEL_NAME_OR_PATH='../pretrained_models/bert_base_cased'

其中MODEL_TYPE 对应的是用的什么transformer预训练模型，例如bert，albert，roberta MODEL_NAME_OR_PATH对应的是这个预训练模型的权重文件的存放地址

blink7-lab commented 3 years ago

好的十分感谢！还有一个小问题如果我想使用electra的话electra-base-discriminator和electra-base-generator是需要都放在pretrained_models中使用吗

liuyukid commented 3 years ago

这个不用的，只要MODEL_NAME_OR_PATH的路径写对了就行，例如/pretrained_models/electra/electra-base-discriminator，而不应该是/pretrained_models/electra/， MODEL_NAME_OR_PATH这个目录下应该要有pytorch_model.bin这个文件的

blink7-lab commented 3 years ago

ok我明白了，谢谢！！！

bubbleqi commented 3 years ago

为什么我的路径按照这个写了还是一直加载不进来呢 一直是这个错误OSError: Can't load 'pretrained_models/bert-base-uncased/,'