The plot summary function works fine but could need a facelift, e.g., joint plotting of multiple tags on top of each other (training loss and validation loss) in different colors. And probably many other small things to improve the UX.

Additionally, it would be great to have a short how-to-guide for this function.