in the task of deraining, does it only take three days for both DGUNet and DGUNet+ networks to train on two NVIDIA Tesla V100 GPUs?

MC-E / Deep-Generalized-Unfolding-Networks-for-Image-Restoration

Accepted by CVPR 2022

126 stars 26 forks source link

in the task of deraining, does it only take three days for both DGUNet and DGUNet+ networks to train on two NVIDIA Tesla V100 GPUs? #4

Open dante1024 opened 2 years ago

dante1024 commented 2 years ago

Thanks for the great work! Here are some details I want to ask. You mentioned in your paper that it takes three days for the model to train on two NVIDIA Tesla V100 GPUs. Because I am only interested in deraining, in the task of deraining, does it only take three days for both DGUNet and DGUNet+ networks to train on two NVIDIA Tesla V100 GPUs?

MC-E commented 2 years ago

Yes, DGUNet+ takes more training time.