lucidrains / med-seg-diff-pytorch

Implementation of MedSegDiff in Pytorch - SOTA medical segmentation using DDPM and filtering of features in fourier space
MIT License
209 stars 28 forks source link

fine-tunning #16

Open Alan-Py opened 1 year ago

Alan-Py commented 1 year ago

Can someone share your pt file,I want to do fine tunning. My loss is too big.