Thanks for the reply.
Yeah, sure. I am referring to the prediction process once the serialized model (say, a DNN model) is loaded.
I am not sure whether pytorch serving supports extra logic injection between each of the neural network layer (during the prediction process). I am suspecting that JIT does not provide such API. Does it require fundamental changes to the source code to enable such a feature?
Best,