Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "c25582d5090cfdcf5f618df4733feecc7a7653b6"
Commit 002cb170 authored by Neel Kant's avatar Neel Kant
Browse files

Correct arguments

parent 183ad176
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
megatron/arguments.py
View file @ 002cb170
......@@ -322,8 +322,10 @@ def _add_data_args(parser):
help='Path to combined dataset to split.')
group.add_argument('--titles-data-path', type=str, default=None,
help='Path to titles dataset used for ICT')
group.add_argument('--hash-data-path', type=str, default=None,
help='Path to pickled HashedIndex data structure')
group.add_argument('--block-data-path', type=str, default=None,
help='Path to pickled BlockData data structure')
group.add_argument('--block-index-path', type=str, default=None,
help='Path to pickled data structure for efficient block indexing')
group.add_argument('--split', type=str, default='969, 30, 1',
help='Comma-separated list of proportions for training,'
' validation, and test split. For example the split '
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment