From 49759c0cda29ab614b81e0869972c99f2edba7aa Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Mon, 23 Nov 2020 15:02:59 -0500 Subject: [PATCH] Document new training argument --- src/transformers/training_args.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py index ff22be4db4..78990cc879 100644 --- 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(