diff --git a/train.py b/train.py
index 583f711..1d60115 100644
--- a/train.py
+++ b/train.py
@@ -55,7 +55,7 @@ train_examples, val_examples = train_test_split(
 # validation examples can be something like templated sentences
 # that maintain the same distance as the cities (same context)
 # should probably add training examples like that too if needed
-BATCH_SIZE = 48
+BATCH_SIZE = 16 * 16
 num_examples = len(train_examples)
 steps_per_epoch = num_examples // BATCH_SIZE