keras backend 報錯OverflowError

File "mtrand.pyx", line 1289, in mtrand.RandomState.uniform (numpy/random/mtrand/mtrand.c:19267)
OverflowError: Range exceeds valid bounds

是的!之前我們在~/.keras/keras.json里改的theano它還是不長記性!
從異常跑出的棧里看是numpy的random函數有越界,
解決:
文件前面初始化設置:
from keras import backend
backend.set_image_dim_ordering('th')

if "image_dim_ordering": is "th" and "backend": "theano", your input_shape must be (channels, height, width)
if "image_dim_ordering": is "tf" and "backend": "tensorflow", your input_shape must be (height, width, channels)

用tensorflow+gpu的時候,在代碼里添加這個,注意input此時變成了(samples,row,col,channels)的格式

from keras import backend
import tensorflow
sess=tensorflow.Session()
backend.set_session(sess)
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容