Document new training argument

49759c0c · Sylvain Gugger · 1cd9be2a · 49759c0c
Commit 49759c0c authored Nov 23, 2020 by Sylvain Gugger
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/transformers/training_args.py src/transformers/training_args.py +3 -0

No files found.
--- a/src/transformers/training_args.py
+++ b/src/transformers/training_args.py
@@ -186,6 +186,9 @@ class TrainingArguments:
            - :obj:`True` if :obj:`metric_for_best_model` is set to a value that isn't :obj:`"loss"` or
              :obj:`"eval_loss"`.
            - :obj:`False` if :obj:`metric_for_best_model` is not set, or set to :obj:`"loss"` or :obj:`"eval_loss"`.
+        model_parallel (:obj:`bool`, `optional`, defaults to :obj:`False`):
+            If there are more than one devices, whether to use model parallelism to distribute the model's modules
+            across devices or not.
    """

    output_dir: str = field(