problems for testing rehistoGAN

[zizixixi]

Hello!

I want to test rehistoGAN using the pretrained pth file you provided. But where should I put the pt file? And where to state the file path for model loading? I can't figure it out from the code.

Thanks a lot!

[svillaza]

Hi!

You must put the pre-trained model in the folder "models". Example

[mahmoudnafifi]

Thanks, svillaza!