Hello,
I am trying to figure out how much RAM is used while running a trained model on a single input like:
model = #model definition and loading trained weights
x = image #load image
y = model(x) #run model on image
I want to basically get the memory usage of this script