Reduce query BS

pomonam · Mar 13, 2024 · 2894d97 · 2894d97
1 parent 935ec78
commit 2894d97
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.",
     )