我在台式机上运行Keras模型,它在笔记本电脑上运行得很顺利。但是在桌面,它给了我以下错误:
File "mtrand.pyx", line 1252, in mtrand.RandomState.uniform (numpy/random/mtrand/mtrand.c:12988)
OverflowError: Range exceeds valid bounds
我的编辑器显示错误发生在这一行:
model.add(Dense(128, activation='relu'))
这可能是由于笔记本电脑和台式机之间的Keras版本不同。请参阅Keras issue #2681。修复可能是更新input_shape()
方法调用,其参数已更改,如下所示:
input_shape=(IMAGE_HEIGHT, IMAGE_WIDTH,1)
如果这不起作用,您可以尝试更改图像的维度顺序:
from keras import backend as K
K.set_image_dim_ordering('th')
据报道,这两种解决方案都适用于遇到此问题的用户。