Keras Maxpooling2d layer gives ValueError

后端 未结 7 1873
难免孤独
难免孤独 2021-02-07 00:10

I am trying to replicate VGG16 model in keras, the following is my code:

model = Sequential()
model.add(ZeroPadding2D((1,1),input_shape=(3,224,224)))
model.add(C         


        
7条回答
  •  忘掉有多难
    2021-02-07 00:43

    Adding dim_ordering solved error for me:

    model.add(MaxPooling2D(pool_size=(2, 2), dim_ordering="th"))
    

提交回复
热议问题