ge-xing / Diff-UNet

Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation. (using diffusion for 3D medical image segmentation)
Apache License 2.0
155 stars 22 forks source link

How long the training time totaly? #8

Open shenhuqiji opened 1 year ago

shenhuqiji commented 1 year ago

According to your code, the diffusion model is not a faster version with large steps. How long the training time you cost?

920232796 commented 1 year ago

Training is fast, which is only one step computation process. But the testing takes 10 steps, so it's a little slow .