Thanks I must have been vague since you got me wrong.
to be more precise, I’m looking for a way to get an updated definition(i.e. latest representation of the model as is) based on the latest changes to the model structure, not the initial model definition.
For example during pruning some featuremaps are pruned and thus some layers that initially had x number of feature maps, now have x’ number. when the model is saved, the state_dicts are saved and thus I still need to find a (sane) way to also extract the model definition and save it along with the weights.
What are my options here?
I hope this makes it clearer.
Thanks a lot in advance