Reduce BS for imagenet

pomonam · Mar 12, 2024 · 6b083d8 · 6b083d8
1 parent a618164
commit 6b083d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/glue/train.py b/examples/glue/train.py
@@ -13,7 +13,7 @@
 
 
 def parse_args():
-    parser = argparse.ArgumentParser(description="Train classification models on MNIST datasets.")
+    parser = argparse.ArgumentParser(description="Train text classification models on GLUE datasets.")
 
     parser.add_argument(
         "--dataset_name",

diff --git a/examples/imagenet/analyze.py b/examples/imagenet/analyze.py
@@ -33,7 +33,7 @@ def parse_args():
     parser.add_argument(
         "--batch_size",
         type=int,
-        default=1024,
+        default=512,
         help="Batch size for compute factors and scores.",
     )
     parser.add_argument(