tensorflow - 如何在 Keras 中设置内核初始化程序 (glorot_uniform) 的种子值

Question

我想在 Keras 中设置 glorot_uniform 内核初始化程序的种子值。

model.add(Dense(50, input_dim=self.state_size, activation='relu', kernel_initializer='glorot_uniform(seed=0)'))

当我使用上面的代码时，错误信息如下。

ValueError: Unknown initializer: glorot_uniform(seed=0)

如果我像下面这样删除“(seed=0)”

 model.add(Dense(50, input_dim=self.state_size, activation='relu', kernel_initializer='glorot_uniform'))

不用设置种子值也能很好地工作。

如何设置种子值？

Answer 1

Keras 可以使用字符串和函数作为初始化器的参数。这些字符串仅使用初始化程序的默认选项。为您的 FC 层尝试这行代码:

from keras import initializers
model.add(Dense(50, input_dim=self.state_size, activation='relu', kernel_initializer=initializers.glorot_uniform(seed=0)))

这里有初始化器的文档: https://keras.io/initializers