我想在我的代码中使用 Intel Extension for Pytorch 来提高整体性能。参考此 GitHub ( https://github.com/intel/intel-extension-for-pytorch ) 进行安装。
目前,我正在尝试一个拥抱人脸摘要 PyTorch 示例(https://github.com/huggingface/transformers/blob/master/examples/pytorch/summarization/run_summarization.py)。下面是用于训练的训练器 API。
# Initialize our Trainer
trainer = Seq2SeqTrainer(
model=model,
args=training_args,
train_dataset=train_dataset if training_args.do_train else None,
eval_dataset=eval_dataset if training_args.do_eval else None,
tokenizer=tokenizer,
data_collator=data_collator,
compute_metrics=compute_metrics if training_args.predict_with_generate else None,
)
我不知道在此代码中启用 Ipex。谁能帮我这个?
提前致谢!