diff --git a/configs/grids/lp_grid.yaml b/configs/grids/lp_grid.yaml
index af93730..0685134 100644
--- a/configs/grids/lp_grid.yaml
+++ b/configs/grids/lp_grid.yaml
@@ -7,9 +7,9 @@ grid:
   optimization.optimizer: [adamw]
   optimization.lr: [0.01, 0.05, 0.001]
   optimization.weight_decay: [5.0e-4, 0]
-  optimization.momentum: [0.9, 0.99, 0.5]
+  optimization.momentum: [0.9, 0.5]
   optimization.lr_schedule: [cosine, step]
-  data.batch_size: [32, 64, 128, 256]
+  data.batch_size: [64, 128, 256]
   meta.representation_type: [last_avgpool, last4_avgpool_concat]
   meta.head_type: [linear, bn_linear]