Silence logging

This commit is contained in:
Nicolas Mowen 2025-05-29 14:52:15 -06:00
parent 4066201ad3
commit e3af887927
2 changed files with 4 additions and 1 deletions

View File

@ -227,6 +227,9 @@ ENV OPENCV_FFMPEG_LOGLEVEL=8
# Set HailoRT to disable logging
ENV HAILORT_LOGGER_PATH=NONE
# TensorFlow error only
ENV TF_CPP_MIN_LOG_LEVEL=3
ENV PATH="/usr/local/go2rtc/bin:/usr/local/tempio/bin:/usr/local/nginx/sbin:${PATH}"
# Install dependencies

View File

@ -87,7 +87,7 @@ def train_classification_model(model_dir: str) -> bool:
f.write(f"{class_name}\n")
# train the model
model.fit(train_gen, epochs=EPOCHS)
model.fit(train_gen, epochs=EPOCHS, verbose=0)
# convert model to tflite
converter = tf.lite.TFLiteConverter.from_keras_model(model)