Preservation of weight issues

[jyfan1104]

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

[nickk124]

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

[jyfan1104]

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

[nickk124]

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]

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