bigger batch size

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