search

iSEE-Laboratory / DiffUIR

The official implementation of the paper of CVPR2024: Selective Hourglass Mapping for Universal Image Restoration Based on Diffusion Model
MIT License
103 stars 7 forks source link

Batch Size Setting #20

Closed henlyta closed 3 weeks ago

henlyta commented 3 weeks ago

I am reaching out for clarification regarding the batch size setting in your paper. In the paper, you mention a batch size of 10 for training, while the provided code sets it to 2. Could you please advise on which setting would be more appropriate to achieve the results discussed in your work? Thank you very much for your time and help!

zhengdian1 commented 3 weeks ago

please look our code carefully !!! We never use batch size 2.