Tensorflow v1.4:Eager模式

时间:2017-11-09 15:05:41

标签: python tensorflow keras eager

我知道Eager模式是夜间构建的一个新的alpha功能,但它还不完美,但我不知道是否有任何tf.keras解决方法可以解决这个问题。

错误Layer.input not supported in Eager mode.在块上触发

model = tf.keras.models.Sequential()
model.add(tf.layers.Dense(2, input_shape = (None, 1)))
model.add(tf.layers.Dense(units = 1))
model.compile(optimizer = "sgd", loss = "mean_squared_error")

我对keras或keras tensorflow API一无所知,我想知道是否有办法避免使用keras技术Layer.input以便保持在Eager模式。根据tf.Eager文档中的教程,我已确认model = tf.layers.Dense(1)有效,但我不知道如何添加其他图层。

非常感谢任何帮助。

修改 从tensorflow v1.10开始,kera在急切模式下受支持。

1 个答案:

答案 0 :(得分:1)

热切执行尚不支持Keras 11.6.0,但Keras层是。这意味着虽然您还不能使用Model,但您可以自己组合图层。请参阅the user guide

希望有所帮助。