search

pfriedri / wdm-3d

PyTorch implementation for "WDM: 3D Wavelet Diffusion Models for High-Resolution Medical Image Synthesis" (DGM4MICCAI 2024)
https://pfriedri.github.io/wdm-3d-io
MIT License
50 stars 5 forks source link

Train time #5

Closed chenyu1330 closed 2 months ago

chenyu1330 commented 2 months ago

Thank you very much for your work! I would like to ask how long it took you to train MODEL: WDM (U-Net) 128 x 128 x 128.

pfriedri commented 2 months ago

@chenyu1330 the training took about 10 days (on a single A100 GPU)