diff --git a/mammoth/opts.py b/mammoth/opts.py index dfdc7858..bbbaafcd 100644 --- a/mammoth/opts.py +++ b/mammoth/opts.py @@ -617,7 +617,13 @@ def _add_train_general_opts(parser): choices=["sents", "tokens"], help="Batch grouping for batch_size. Standard is sents. Tokens will do dynamic batching", ) - group.add('--pad_to_max_length', '-pad_to_max_length', action='store_true') + group.add( + '--pad_to_max_length', + '-pad_to_max_length', + action='store_true', + help='Pad all minibatches to max_length instead of to the length of the longest sequence in the minibatch. ' + 'Using this together with batch_type=sents results in tensors of a fixed shape.' + ) group.add('--max_length', '-max_length', type=int, default=None, help='Maximum sequence length.') group.add( '--task_distribution_strategy',