每次测试时结果不同

Algolzw / image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.

https://algolzw.github.io/ir-sde/index.html

MIT License

521 stars 39 forks source link

每次测试时结果不同 #65

Open MrWan001 opened 7 months ago

MrWan001 commented 7 months ago

作者您好，我用训练出的同一个模型权重，在每一次运行test.py测试时得到的结果不同，请问这是什么原因，我应该如何得到一个固定的结果期待您的回复。

Algolzw commented 7 months ago

因为我们方法每一步都加了随机噪声所以结果会有不同，你可以尝试设置numpy和pytorch的random seed来固定一下。

MrWan001 commented 6 months ago

好的，感谢您对每一条问题的耐心回复。