about the size of tgt_pose when inference

[gouchaonijiao]

Hello! Your work is very impressive,but I wonder why the image and tgt_tensor are scaled to 256*256? And how I get other size of tgt_tensor like (256,512). Wish your reply!