icon-lab / SynDiff

Official PyTorch implementation of SynDiff described in the paper (https://arxiv.org/abs/2207.08208).
Other
244 stars 37 forks source link

The issue of training speed #51

Open icecreamlovepudding opened 2 weeks ago

icecreamlovepudding commented 2 weeks ago

Hi,your work is excellent,and I want to try on my own dataset(CBCT-CT) However, during the training process, it would take almost a week for only 50 epochs! My training dataset consisted of 500 images and the validation dataset consisted of 20 pairs. The hyperparameters are set according to your article.The GPU is NIVIDIA GeForce RTX2080. I really want to know the reason.Look forward to your replay.