Skip to content

GitLab

Commit 78071e0f authored by Myle Ott's avatar Myle Ott
Browse files

Fix type of c10d bucket size

parent 862cad11
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
fairseq/options.py
View file @ 78071e0f
...@@ -187,7 +187,7 @@ def add_distributed_training_args(parser): ...@@ -187,7 +187,7 @@ def add_distributed_training_args(parser):
help='which GPU to use (usually configured automatically)') help='which GPU to use (usually configured automatically)')
group.add_argument('--no-c10d', action='store_true', group.add_argument('--no-c10d', action='store_true',
help='don\'t use c10d distributed backend') help='don\'t use c10d distributed backend')
group.add_argument('--c10d-bucket-cap-mb', default=150, metavar='MB', group.add_argument('--c10d-bucket-cap-mb', default=150, type=int, metavar='MB',
help='bucket size for c10d backend') help='bucket size for c10d backend')
return group return group
......
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