search

cszn / SCUNet

Practical Blind Denoising via Swin-Conv-UNet and Data Synthesis (Machine Intelligence Research 2023)
https://link.springer.com/article/10.1007/s11633-023-1466-0
Apache License 2.0
619 stars 63 forks source link

Training time of SCUNet #12

Closed ncbnwklrt closed 1 year ago

ncbnwklrt commented 2 years ago

you have mention in paper that

It takes about three days to train a denoising model on four NVIDIA RTX 2080 Ti GPUs.

however, in my case, it takes about 5 days to train on 4 V100 GPUs for 1200000 iterations and pics are crop into 544*544. It is much longer than 3 days. How can i speed up? thanks.

zhongqiu1245 commented 1 year ago

@ncbnwklrt Hello, could you release your training code? Thx!