Graph Packing with VTA and Unet

Hello, I have shared the model and the VTA code imported into the model on the web disk,this is link I only changed the import network and graph_pack, not the rest of the test section. I would appreciate it if you could help debug。Since it is the first time for me to contact VTA, this problem has been bothering me for a long time and has not been solved yet. I am looking forward to your help. Thank you.