pytorch 如何解决拥抱脸解析器的这个错误

fwzugrvs  于 2023-03-02  发布在  其他
关注(0)|答案(1)|浏览(189)

我正在尝试运行带有参数解析器的拥抱脸训练器(pytorch)。

if __name__ == '__main__':
parser = HfArgumentParser(TrainingArguments)
parser.add_argument('--model_name_or_path', type=str, required=True)
.
.
.   
.

training_args = parser.parse_args()

print('args', training_args)

os.makedirs(training_args.output_dir, exist_ok=True)

random.seed(training_args.seed)

set_seed(training_args.seed)

dataset_train = ...

.
.
.

trainer = Trainer(
    model=model,
    args=training_args,
    train_dataset=train_dataloader,
    eval_dataset=val_dataloader)

trainer.train()

我收到以下错误:

Traceback (most recent call last):
File "main.py", line 250, in <module>
eval_dataset=val_dataloader)
File "C:\User\transformer\lib\site- 
packages\transformers\trainer.py", line 316, in __init__
log_level = args.get_process_log_level()
AttributeError: 'Namespace' object has no attribute 'get_process_log_level

关于此错误以及如何解决此错误的任何想法

6jjcrrmo

6jjcrrmo1#

首先检查一下这是否适合您:

from transformers import TrainingArguments

args = TrainingArguments(output_dir='./')
args.get_process_log_level()

[out]:

20

如果不匹配,那么很可能是C:\User\transformer\lib\site-packages\transformers上的transformers版本与您的Trainer脚本不匹配。然后尝试升级您的变形金刚版本pip install -U transformers

相关问题