代码之家  ›  专栏  ›  技术社区  ›  Stupid420

keras层中的尺寸失配

  •  0
  • Stupid420  · 技术社区  · 7 年前

    我正在建立一个小型CNN+RNN网络如下。

    batch_size=2048
    
    
    visible = Input(shape=(batch_size,251))
    
    embed=Embedding(58, 50)(visible)
    x1=keras.layers.Conv1D(92,50,activation='relu')(embed)
    x2=keras.layers.GRU(224, return_sequences=True)(x1)
    
    x3=keras.layers.GRU(284)(x2)
    
    predictions=Dense(1, activation='sigmoid')(x3)
    

    如何使不同层之间的尺寸兼容性合适。我只是通过运行这段代码得到以下错误。

    ValueError: Input 0 is incompatible with layer conv1d_4: expected ndim=3, found ndim=4
    
    1 回复  |  直到 6 年前
        1
  •  0
  •   Stupid420    7 年前

    visible = Input(shape=(251,))