tensorflow版本错误

跑程序出现ValueError: Negative dimension size caused by subtracting 2 from 1 for ‘max_pooling2d_2/MaxPool’ (op: ‘MaxPool’) with input shapes: [?,1,128,128].

参考:https://github.com/keras-team/keras/issues/3945

主要是因为TensorFlow版本。0.10使用“TF”,0.11使用“th”。也就是在所有模型里的MaxPool参数更改,即model.add(MaxPooling2D(pool_size=(2, 2), dim_ordering=“th”))

你可能感兴趣的:(深度学习)