如何使用Keras在CNN中执行前馈传播?

时间:2017-11-11 01:09:29

标签: neural-network keras conv-neural-network

我想使用Keras在CNN上执行前馈传播。我正在尝试使用我自己的优化器训练CNN,我无法在Keras的优化器文件中使用。我的渐变自由优化器。我不希望任何内置使用。

1 个答案:

答案 0 :(得分:0)

I found answer to this question. We just have to make model non trainable.

import numpy as np
import keras

x = keras.layers.Input(shape=(3,))
y = keras.layers.Dense(5)(x)

model = keras.models.Model(x, y)
model.trainable = False
model.compile(optimizer='rmsprop', loss='mse')

x = np.random.random((10, 3))
y = np.random.random((10, 5))
model.fit(x, y, epochs=10)