budai4medtech / xfetus

xfetus -- A library for ultrasound fetal imaging synthesis using techniques from GANs, transformers, and diffusion models.

6 stars 1 forks source link

32 - style swin gan #33

Closed mxochicale closed 1 year ago

mxochicale commented 1 year ago

Addressing #32

seansunn commented 1 year ago

CUDA memory > Hi Migual, does the CUDA out of memory error occurred during training or during inferencing? If it's during training, I can implement gradient accumulation to see if it will solve this issue.

checkpoint > the checkpoint is not uploaded. Do you want it to be integrated in the package?