Added log level parameter

Nolte-Group · Jan 6, 2025 · 540757d · 540757d
1 parent a48f78b
commit 540757d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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)