内存中出现Numpy错误.检验文件



我在台式机上运行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')

据报道,这两种解决方案都适用于遇到此问题的用户。

相关内容

  • 没有找到相关文章

最新更新