文章/答案/技术大牛

发布

社区首页 >问答首页 >胶囊网络和可重复性

问胶囊网络和可重复性
EN

Stack Overflow用户

提问于 2019-10-24 04:51:41

回答 1查看 121关注 0票数 0

我目前在使用胶囊网络(Keras版本: CapsNet)时遇到了困难。

每次我连续运行2-5个以上的预测(在循环的一侧)，结果会有很大的不同。我试着去改变这样的事情。我也把优化器从ADAM改成了SGD --但我就是不能让它100 %稳定，从而再一次重现一个给定的运行。

如何才能使CapsNet在每次运行时都100%可重现？

发布于 2019-10-24 05:03:39

这个问题的答案是冗长而复杂的。这里有a blog post that goes into much more detail than I can here，但我将尝试捕获高层次的要点。

在运行python程序之前将PYTHONHASHSEED环境变量设置为0。
如果您在GPU上运行计算，由于浮点舍入，这可能会导致不可重现的结果。您可以像以前一样，通过将CUDA_VISIBLE_DEVICES环境变量设置为空字符串来禁用它并在CPU上运行所有操作。

CUDA_VISIBLE_DEVICES="" PYTHONHASHSEED=0 python your_program.py

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/58530832

复制

相似问题

问胶囊网络和可重复性EN