So far the learned prototype patches are just visualised at the end of the training and not stored (otherwise we will end up with 2000 saved png images).
We need to decide how we want to handle this.
Furthermore the visualization is done only if we run the prototype explanation first during the evaluation of the model.