search

KidsWithTokens / MedSegDiff

Medical Image Segmentation with Diffusion Model
MIT License
976 stars 145 forks source link

Bugs that cannot support fp16 training #135

Open RoboticsZhang opened 9 months ago

RoboticsZhang commented 9 months ago

After setting use_fp16 = True, some bugs occur: image It seems that not all of the models and weights are converted to fp16? Looking forward to your reply, thank you