I saw the link
https://pytorch.org/tutorials/recipes/mobile_perf.html#setup 2
how to optimize the model. But I got stuck in model preparation.
How can I use my model to replace the code below?

image997×751 31.1 KB