Thanks. It’s just that the API I’m building needs Pytorch to save and load the whole thing at once.
I’m building an explanation API that takes in a model, and an image and produces a Class activation map( CAM).
I want the API to work for any model that is supplied, this works for tf models, since you can save and load them quite simply,  but not for Pytorch models, because I need to instantiate the model definition “before” I fire my API up, which is very inconvenient.