DifFace
DifFace copied to clipboard
DifFace: Blind Face Restoration with Diffused Error Contraction (PyTorch)
Thank for sharing your great work!! : ) We have tried to training DifFace with SwinIR diffused estimator and got an error that self.loss_fun(line 492, 495 in trainer.py) is not...
Could you add a device number to the command line parameters?
followed the example code from the [huggingface model card](https://huggingface.co/OAOA/DifFace) but got the error then tried the below with the same error ``` from diffusers import DiffusionPipeline model_id = "OAOA/DifFace" #...
I retrained the Train diffused estimator (SwinIR) stage and found that the saved weights are dict_keys(['iter_start','log_step','log_step_img','state_dict']) and not SwinIR's weights, so the inference reports a load model error. Would like...
I met this problem when I tried to run the command CUDA_VISIBLE_DEVICES=0 python main_sr.py --cfg_path configs/training/swinir_ffhq512.yaml --save_dir save_dir_1 Could someone help me solve it?
I met this problem when I tried to run the command `CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun --standalone --nproc_per_node=4 --nnodes=1 main_diffusion.py --gpu_id 0123 --cfg_path configs/training/diffusion_ffhq512.yaml --save_dir myfolder`. Could someone help me solve it?
Have you ever tried to decrease the lr(the default value is 1e-4) for the training procedure of diffusion model? Since the loss converges very quick(less than 1W iters), but the...
This file is not available for model training
adjust_lr() missing 1 required positional argument: 'ii'adjust_lr() missing 1 required positional argument: 'ii'. In trainer_py trainer.py", line 255, in train self.adjust_lr()self.adjust_lr() Please correct me if I am wrong? What to...
python.exe D:\develop\DifFace\inference_difface.py --in_path D:\develop\DifFace\testdata\whole_imgs --out_path D:\data --gpu_id 0 Setting random seed 20000 Loading from ./weights/diffusion/iddpm_ffhq512_ema500000.pth... Loaded Done C:\Users\******\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass...