Phase Retrieval settings for Resample on FFHQ

[seulaugues]

Hi,

I am trying to run the phase retrieval task from your DiffStateGrad code on the FFHQ dataset. I used the default configuration provided in configs/tasks/phase_retrieval_config.yaml with the resample sampler.

However, the reconstruction quality is much lower than expected — the PSNR I obtain is less than 15. I am not sure if I am using the intended settings for this task.

Could you please clarify:

1. Which configuration files and hyperparameters you typically use for phase retrieval in your experiments (e.g., latent model, configurations, etc.)?
2. Any special adjustments for resample sampler when using phase retrieval?
3. Are there recommended settings for FFHQ specifically?

Thank you very much for your help!