@hjiang In addition, I would like to ask, can Nn. upsample be implemented in VTA? In the process of running Unet, I suspected that the error was caused by Upsample or CAT. Have you ever implemented this layer? A detailed description can be found in another help I wrote in Can Upsample be implemented on VTA in graph_pack?. Do you have any thoughts or suggestions? Looking forward to your reply. Thank you very much