From 540757d25cd22d3f6fb9c1d6ab4ecd3d04a99fbd Mon Sep 17 00:00:00 2001 From: maelstrom Date: Mon, 6 Jan 2025 01:35:27 -0500 Subject: [PATCH] Added log level parameter --- train.py | 3 ++- train_cpu.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 11b0b8c..a447ddd 100644 --- a/train.py +++ b/train.py @@ -42,6 +42,7 @@ # training parameters BATCH_SIZE = params["batch_size"] EPOCHS = params["epochs"] + LOG_LEVEL = params["log_level"] del params def trim(dataframe, column): @@ -139,7 +140,7 @@ def main(): start = time.time() model.fit(x=train_set[0], y=train_set[1], validation_data=(validation_set[0], validation_set[1]), - batch_size=BATCH_SIZE, epochs=EPOCHS) + batch_size=BATCH_SIZE, epochs=EPOCHS, verbose=LOG_LEVEL) end = time.time() print("Training time: ", end - start) diff --git a/train_cpu.py b/train_cpu.py index c1dbe24..80d897c 100644 --- a/train_cpu.py +++ b/train_cpu.py @@ -43,6 +43,7 @@ # training parameters BATCH_SIZE = params["batch_size"] EPOCHS = params["epochs"] + LOG_LEVEL = params["log_level"] del params def trim(dataframe, column): @@ -127,7 +128,7 @@ def main(): start = time.time() model.fit(x=train_set[0], y=train_set[1], validation_data=(validation_set[0], validation_set[1]), - batch_size=BATCH_SIZE, epochs=EPOCHS) + batch_size=BATCH_SIZE, epochs=EPOCHS, verbose=LOG_LEVEL) end = time.time() print("Training time: ", end - start)