From 8c5001166d2a4725a8f12f7f771b74ffa3d5dbfb Mon Sep 17 00:00:00 2001 From: Jordon Brooks Date: Wed, 26 Jul 2023 00:12:36 +0100 Subject: [PATCH] Changed batch size --- train_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/train_model.py b/train_model.py index 9127ec8..bbef678 100644 --- a/train_model.py +++ b/train_model.py @@ -9,9 +9,9 @@ from tensorflow.keras.callbacks import EarlyStopping print(tf.config.list_physical_devices('GPU')) # Constants -BATCH_SIZE = 16 -EPOCHS = 1 -TRAIN_SAMPLES = 1 +BATCH_SIZE = 8 +EPOCHS = 5 +TRAIN_SAMPLES = 5 def load_list(list_path): with open(list_path, "r") as json_file: