Getting an error with a multi-input Keras model related to the input dimension

后端 未结 1 1332
予麋鹿
予麋鹿 2021-01-27 11:32

I have a multi input Keras model. Here the inputs:

[,
 

        
1条回答
  •  粉色の甜心
    2021-01-27 11:43

    try

    train_x_list = [tf.squeeze(tx) for tx in tf.split(train_x, num_or_size_splits=train_x.shape[0], axis=0)]
    

    it will produce a list of tensors with training data split along dimension 0. Then use your second solution, feeding the list to fit().

    0 讨论(0)
提交回复
热议问题