Training using multicore on Centos7 w/ AMD EPYC 7513 32-Core Processor

Hi,

I’m trying to run the following code on my server using all cores of AMD EPYC 7513 32-Core Processor but the process uses just 1 core at the time and I don’t know why…

Someone can help me?

Regards

output_dir = f’./dialogue-summary-training-{str(int(time.time()))}’

training_args = TrainingArguments(
output_dir=output_dir,
learning_rate=1e-15,
num_train_epochs=1,
weight_decay=0.01,
logging_steps=1,
max_steps=500,
use_cpu=True
)

trainer = Trainer(
model=original_model,
args=training_args,
train_dataset=tokenized_datasets[‘train’],
eval_dataset=tokenized_datasets[‘validation’],
)

trainer.train()