segmentation-guided-diffusion icon indicating copy to clipboard operation
segmentation-guided-diffusion copied to clipboard

Published 20 hours ago • verified publisher icon mazurowski-lab

Preservation of weight issues

Open jyfan1104 opened this issue 11 months ago • 4 comments

image Dear Author, I'm having a problem with my training, have you ever had this problem?

jyfan1104 avatar Jan 11 '25 08:01 jyfan1104

Thanks for using our code! This bug is a known issue: please see the second bullet point under “Troubleshooting/Bugfixing” in the main README here: https://github.com/mazurowski-lab/segmentation-guided-diffusion?tab=readme-ov-file#troubleshootingbugfixing

nickk124 avatar Jan 13 '25 16:01 nickk124

Thank you very much, during training I set the value of num_epochs to 2000, why does the training end when it reaches 500

jyfan1104 avatar Jan 14 '25 12:01 jyfan1104

That's strange, I'm not sure why that's happening. Can you please provide the full command that you're using to run the code?

nickk124 avatar Jan 14 '25 15:01 nickk124

Ok, I see. Is the model training all 2000 epochs but just not saving past 500, or is it only training 500?

nickk124 avatar Jan 14 '25 16:01 nickk124