diff --git a/orttraining/orttraining/python/ort_trainer.py b/orttraining/orttraining/python/ort_trainer.py index b350e3c578c77..0336a6c8538b3 100644 --- a/orttraining/orttraining/python/ort_trainer.py +++ b/orttraining/orttraining/python/ort_trainer.py @@ -574,9 +574,6 @@ def __init__(self, model, loss_fn, model_desc, training_optimizer_name, map_opti gradient_accumulation_steps: number of training steps to accumulate gradients before averaging and applying them. Defaults to 1. - postprocess_model: a callable to postprocess the ONNX model that is - converted from PyTorch. - Defaults to None. world_rank: rank id used for distributed training. Defaults to 0. world_size: number of ranks participating in distributed training.