if args.model_name == 'bert': self.tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased') self.input_size = 768 if args.model_name == 'bert': 这里的input-size,需要根据什么来确定值 所以是因为这个模型bert本来就应该是768,而不用根据具体的应用来选择这个值?self.tokenizer = AutoTokeniz...
python train.py --model_name_or_path bert-base-uncased 重新运行修改后的命令: 在修改命令并包含了所有必需的参数后,重新运行脚本。 检查是否还会出现相同的错误信息。如果错误信息消失,且脚本能够正常运行,那么说明问题已经解决。 综上所述,解决这个问题的关键是确保在运行 train.py 脚本时提供了所有必需的命...
importtorchfromtransformersimportAutoModel,AutoConfigfromhuggingface_hubimporthf_hub_downloadmodel_id="bert-base-uncased"num_classes=2model_class=AutoModelstate_dict=torch.load(hf_hub_download(model_id,revision="main",filename="pytorch_model.bin"))huggingface_config_path=Noneconfig=model_class.from_pre...