I want to only store the layers block and the computation graph.
To start off I want it to work for Sequential models. I want to be able to store the Sequential model architecture without having to store the parameters. Is that possible?
After that, is it generalizable for arbitrary models? I only want to store the shape and types of the pytorch graph. I don’t need the parameters.