训练时改变高斯噪声层的 Keras 标准偏差

Question

在训练期间更改 Keras 中高斯噪声层的标准偏差的最快方法是什么？

目前我在每次迭代时都用调整后的标准差重新加载整个网络，这真的很慢。

提前致谢！

Answer 1

您可以尝试使用 keras 后端变量吗？

from keras import backend as K

self.std=0.5
self.std_var = K.variable(value=std)

构建网络时。

# instantiate
stddev = std_var(0.8)
g = GaussianNoise(stddev)

训练期间（可能在循环内）。

K.set_value(stddev.std_var, <new_std_val>)

试试这个片段，看看它是否有效。