From 2894d9795741bbd86315442b3d371fff5b45724f Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Wed, 13 Mar 2024 02:29:43 -0400 Subject: [PATCH] Reduce query BS --- examples/imagenet/ddp_analyze.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/imagenet/ddp_analyze.py b/examples/imagenet/ddp_analyze.py index 2499190..e9a0c8c 100644 --- a/examples/imagenet/ddp_analyze.py +++ b/examples/imagenet/ddp_analyze.py @@ -53,13 +53,13 @@ def parse_args(): parser.add_argument( "--query_batch_size", type=int, - default=128, + default=64, help="Batch size for computing query gradient.", ) parser.add_argument( "--train_batch_size", type=int, - default=128, + default=256, help="Batch size for computing training gradient.", )