How to inference with the weight obtained by training stage 1?

[Zeyu-Zhu]

Thanks for the great work!

I'm wondering how to run inference_long.py with the weight I obtained by training stage 1 on my only datasets.

I tried to combine ['denoising_unet-xxxpth', 'face_locator-xxxpth', 'imageproj-xxx.pth', 'reference_unet-xxx.pth'] into one file 'net.pth' but when I ran inference_long.py it didn't work.

Could you give me any advice? Looking forward to your reply!

Best