From 03313d3904177203a8f769a2747d539c236f5c7f Mon Sep 17 00:00:00 2001 From: mm Date: Fri, 5 May 2023 01:28:35 +0000 Subject: [PATCH] bigger batch size --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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