From da32d75778659599e8017760c4c367177fcf3ed7 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 31 Oct 2022 18:36:28 +0000 Subject: [PATCH] make_callbacks: display steps, not samples --- aimodel/src/lib/ai/helpers/make_callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/helpers/make_callbacks.py b/aimodel/src/lib/ai/helpers/make_callbacks.py index bfb112a..f5fadd2 100644 --- a/aimodel/src/lib/ai/helpers/make_callbacks.py +++ b/aimodel/src/lib/ai/helpers/make_callbacks.py @@ -24,5 +24,5 @@ def make_callbacks(dirpath, model_predict): filename=filepath_metrics, separator="\t" ), - tf.keras.callbacks.ProgbarLogger() + tf.keras.callbacks.ProgbarLogger(count_mode="steps") # batches ] \ No newline at end of file