saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-20 19:12:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove docstring for removed parameter (#4734)

Browse source
This commit is contained in:
Sergii Dymchenko 2020-08-07 11:43:36 -07:00 committed by GitHub
parent b22091dc91
commit c334b5738e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
orttraining/orttraining/python/ort_trainer.py
View file

@ -574,9 +574,6 @@ class ORTTrainer():
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.