from optimum.graphcore import IPUTrainingArguments 
ipu_arguments = IPUTrainingArguments( 
    f"medical-bert-ner-finetuned", 
    evaluation_strategy = "epoch", 
    learning_rate=2e-5, 
    per_device_train_batch_size=16, 
    per_device_eval_batch_size=16, 
    gradient_accumulation_steps=10, 
    pod_type="pod4", 
    num_train_epochs=3, 
    weight_decay=0.01, 
    dataloader_drop_last=True, 
    logging_steps=10, 
    push_to_hub=False, 
)