使用numpy.array初始化初始化

时间:2016-11-24 23:48:00

标签: python initialization tensorflow autoencoder tflearn

我正在使用tflearn创建自动编码器,现在我想使用RBM中学习的参数(np.array)来初始化权重和偏差。

我尝试了以下方法:

w = tf.get_variable('w1', shape=(784, 256), initializer=tf.constant_initializer(w0))

encoder = tflearn.fully_connected(encoder, 256, activation='sigmoid',
                                  weights_init=w, bias_init=b)

但我总是得到错误

ValueError: If initializer is a constant, do not specify shape.

如果我使用:

w = tf.get_variable('w1', initializer=tf.constant_initializer(w0))

我收到了错误:

    ValueError: Shape of a new variable (w1) must be fully defined, but instead was <unknown>.

有人能提出一些建议吗?

0 个答案:

没有答案