keras 報錯 ValueError: Tensor conversion requested dtype int32 for Tensor with dtype float32: 'Tensor(

 

在服務器上訓練並保存模型，複製到本地以後load_model()報錯:

 ValueError: Tensor conversion requested dtype int32 for Tensor with dtype float32: 'Tensor("embedding_1/random_uniform:0", shape=(5001, 128), dtype=float32)'

 

緣由：服務器和本地的運行環境配置不一樣

 

查看本地keras版本：

In [1]: import keras
Using TensorFlow backend.

In [2]: keras.__version__
Out[2]: '2.0.6'

（或者，用pip freeze列出全部包的版本信息）

而服務器上的keras版本是2.0.3

 

本地pip install 指定版本安裝：

$ sudo pip install Keras==2.0.3