search

Zj-BinXia / DiffIR

This project is the official implementation of 'Diffir: Efficient diffusion model for image restoration', ICCV2023
Apache License 2.0
451 stars 18 forks source link

Finetuning the RealSR pretrained model #20

Closed yxldd closed 10 months ago

yxldd commented 10 months ago

I want to try finetuning the pretrained model to have a better performance on my privative data, but my single gpu is not suitable for the current finetuning program because of the distributed training mode. It's hard for me to deal with the problems on my own. So, how can I adjust my code to run the finetuning program? I would very much like to use DiffIR to solve problems in the field of my research area.

Zj-BinXia commented 10 months ago

you can run "python3 DiffIR/train.py -opt options/train_DiffIRS2_GAN_x4.yml"