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

丢失函数中的“MXNet后端:无法自动推断卷积运算符的形状”

  •  1
  • bobpoekert  · 技术社区  · 8 年前

    我在打电话keras.backend.conv2d在loss函数中,当我试图编译模型时,会得到以下错误:

    MXNet后端:无法自动推断卷积运算符的形状。请提供输入形状。给定输入形状-(无,无,无,无)

    1 回复  |  直到 8 年前
        1
  •  1
  •   Sina Afrooze    8 年前

    通常你只需要添加 input_shape 作为一个论点 Conv2D . 例如:

    model.add(Conv2D(32, (3, 3), padding='same', input_shape=(256,256,3)))