I'd love to see the architecture used (as defined in the Architecture section of VisualML) in the model Summary / Report page somehow.
When you're trying out different architectures you have to revert the design to a specific session each time you want to view the used architecture, which is a pain. And the added hazard of not realizing what session's settings the Design is set to when training a new model.
JUst the contents of build_model() and compile_model() functions would be enough.
