From bd0bd9be199e565f9affa2308e3e7c113944a91d Mon Sep 17 00:00:00 2001
From: qasfb <36480216+qasfb@users.noreply.github.com>
Date: Wed, 30 Aug 2023 17:52:02 +0200
Subject: [PATCH] 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.
---
dinov2/run/submit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dinov2/run/submit.py b/dinov2/run/submit.py
index 68140f3..38c7034 100644
--- a/dinov2/run/submit.py
+++ b/dinov2/run/submit.py
@@ -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",
)
--
GitLab