keras ‘shape_invariants’: ошибка из серверной части tensorflow

#keras

#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
Ошибка значения: форма для while_3 / Merge_2:0 не является инвариантом для цикла. Он входит в цикл с формой (10, 3, 480, 640), но имеет форму (20, 1, 120, 16) после одной итерации. Укажите инварианты формы, используя либо shape_invariants аргумент tf.while_loop, либо set_shape() для переменных цикла.