nouffer/SumasenTrainer
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

removing tensorflow-docs

Browse Source
This commit is contained in:
Mohamed Nouffer
2021-12-15 09:34:59 +05:30
parent f8daa3d810
commit 37408b00ee
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
requirements.txt
View File

@ -109,7 +109,6 @@ tensorboard
tensorboard-data-server tensorboard-data-server
tensorboard-plugin-wit tensorboard-plugin-wit
tensorflow tensorflow
tensorflow-docs
tensorflow-estimator tensorflow-estimator
termcolor==1.1.0 termcolor==1.1.0
terminado==0.12.1 terminado==0.12.1

3
trainer.py
View File

@ -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)],) 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) model_path = os.path.join(st.session_state['output_folder'], model_name)
st.session_state['model_folder'] = st.session_state["output_folder"] 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) model.save(model_path)
return history return history