"vscode:/vscode.git/clone" did not exist on "d1bea9e87ff9d473ce2de9f99fafe2d37af5dfa8"
Commit 78071e0f authored by Myle Ott's avatar Myle Ott
Browse files

Fix type of c10d bucket size

parent 862cad11
......@@ -187,7 +187,7 @@ def add_distributed_training_args(parser):
help='which GPU to use (usually configured automatically)')
group.add_argument('--no-c10d', action='store_true',
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')
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