我想使用Keras在CNN上执行前馈传播。我正在尝试使用我自己的优化器训练CNN,我无法在Keras的优化器文件中使用。我的渐变自由优化器。我不希望任何内置使用。
答案 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)