diff --git a/newsearch.py b/newsearch.py
index 978f59e..89c0ec1 100644
--- a/newsearch.py
+++ b/newsearch.py
@@ -73,7 +73,7 @@ for idx, params in enumerate(search_params):
 python newmain.py fit \
 --seed_everything {s} \
 --data.batch_size {bs} \
---data.train_size 0 \
+--data.train_size 10000 \
 --data.val_size 10000 \
 --model.alpha {a} \
 --model.width {w} \