Implemented weight saving callback in model_training.py

375304d4 · lucas_miranda · e5feb03e · 375304d4
Commit 375304d4 authored Jun 19, 2020 by lucas_miranda
--- a/visualizations/train_viz_app.py
+++ b/visualizations/train_viz_app.py
@@ -59,7 +59,7 @@ fig2 = px.bar(

 fig2.layout.updatemenus[0].buttons[0].args[1]["frame"]["duration"] = 1
 fig2.update_xaxes(showgrid=True)
-fig2.update_yaxes(showgrid=True, range=[0, samples / 1.5])
+fig2.update_yaxes(showgrid=True, range=[0, samples])

 fig2.update_layout({"paper_bgcolor": "rgba(0, 0, 0, 0)"})