My question is similar in some sense so posting it here. I want to save partial state dict and architecture and be able to load it back. Like saving the weights of only encoder and use that as a part of other model.
Thanks!