.summary() on mx.mod.Module

Gluon
1

I have an mx.mod.Module.
I saw that mx.gluon.SymbolBlock(...).summary(input) gives a beautiful description of the model.

Is there any way to cast an mx.mod.Module to a mx.gluon.SymbolBlock ?
Or do you have any other cleaner way just to call .summary() on the mx.mod.Module?

2

mx.viz.print_summary(net) prints a super detailed output (including every tensor operation), which I don’t need.