From 37408b00eebcd08e76adcfa2c37153ae54cb9ad0 Mon Sep 17 00:00:00 2001 From: Mohamed Nouffer Date: Wed, 15 Dec 2021 09:34:59 +0530 Subject: [PATCH] removing tensorflow-docs --- requirements.txt | 1 - trainer.py | 3 ++- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 7fa5aa5..183f161 100644 --- a/requirements.txt +++ b/requirements.txt @@ -109,7 +109,6 @@ tensorboard tensorboard-data-server tensorboard-plugin-wit tensorflow -tensorflow-docs tensorflow-estimator termcolor==1.1.0 terminado==0.12.1 diff --git a/trainer.py b/trainer.py index ec08195..11a58a7 100644 --- a/trainer.py +++ b/trainer.py @@ -88,7 +88,8 @@ def trainSimplaeModel(model, epochs, tran_iterater, test_iterater, model_name, t history = model.fit(tran_iterater, validation_data=test_iterater, epochs=epochs, callbacks=[CustomCallback(text_output, progrss_bar, graph)],) model_path = os.path.join(st.session_state['output_folder'], model_name) st.session_state['model_folder'] = st.session_state["output_folder"] - #st.write(model_path) + st.markdown(f'##### Model output : {model_path}') + #st.write(f'Output folder is : {model_path}') model.save(model_path) return history