Skip to content
Snippets Groups Projects
Unverified Commit bd0bd9be authored by qasfb's avatar qasfb Committed by GitHub
Browse files

Set default number of nodes to 1

We put default number of nodes to 1 in particular so that the linear evaluation with 8 GPUs has a global batch size of 1024, to reproduce the results.
parent ebc1cba1
No related branches found
No related tags found
No related merge requests found
......@@ -45,7 +45,7 @@ def get_args_parser(
parser.add_argument(
"--nodes",
"--nnodes",
default=2,
default=1,
type=int,
help="Number of nodes to request",
)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment