keras' shape_invariants' :tensorflow后端的错误

时间:2016-10-09 12:20:39

标签: keras

对于我的模型,下面是第一层。输入的形状(10,3,480,640)

model.add(TimeDistributed(
Convolution2D(16, 8, 8,subsample=(4, 4), border_mode="same", init="he_normal")))
model.add(Activation("relu"))

我从tensorflow后端收到此错误 ValueError:while_3 / Merge_2:0的形状不是循环的不变量。它进入具有形状(10,3,480,640)的环,但在一次迭代后具有形状(20,1,120,16)。在循环变量上使用tf.while_loop的shape_invariants参数或set_shape()提供形状不变量。

0 个答案:

没有答案