如何获取和设置随机状态(种子)

时间:2018-03-02 14:47:51

标签: tensorflow

情况:我训练了一段时间,然后想要将当前的列车状态保存到磁盘,然后退出。然后我想继续训练。它应该是完全相同的行为,就像我不会退出一样。

为简单起见,我们假设我使用SGD,尽管存储更新程序状态(Adam等)也不是问题。

但是,我不知道如何阅读和存储随机状态。因此,当我下次重新创建图形和新会话时,它将不会继续随机序列(要么我已经确定了它,那么它将在第一次启动时启动,或者它将是随机的)。

那么,我怎样才能读取随机状态?或者是随机种子,如果我稍后用该种子初始化,它将继续使用相同的序列?

如果那是不可能的,也许我可以使用其他随机生成器吗?我发现了tf.contrib.stateless似乎提供了这样的内容。例如。我可以用...像:

tf.contrib.stateless.stateless_random_normal(..., seed=global_step * some_number)

0 个答案:

没有答案