版本:
OS : Ubuntu 14.04 LTS
tensorflow 1.2
keras 2.2.1
在調用softmax函數時出現 TypeError: softmax() got an unexpected keyword argument 'axis' 的錯誤,其主要原因是因為keras在將語法轉換成tensorflow時某些部分沒有對應上,主因應該是版本落差導致,若不透過更新到匹配版本的情況下,可以修改keras 套件中的部分程式碼,使其匹配.
而從錯誤訊息中可以發現,問題檔案為:
/usr/local/lib/python2.7/dist-packages/keras/backend
因此使用編輯器(vim/nano) 編輯 tensorflow_backend.py
找到3458行中的 axis=axis.
並將其移除,修改成果如下圖:
修改完畢後正常情況即可使用.
文章標籤
全站熱搜
留言列表
Revit 二次開發 (2)
{{ article.title }}