I think you can always output the activations using the org_model, or debug in python as you want, and if you think it’s good, then do the tracing and serialization.
If you want to see the intermediates in the traced model, you can still modify the original model and add print stmts etc to debug it.