Add some setting which allows the option to choose (on the fly) how much visualizations the model should generate, with the balance that the more visualizations that are generated, the slower it will train.