I am working on a codebase written in Tensorflow version 1. Entire computations make use of TF sessions rather than Keras API. The codebase contains a ResNet model, on which I a