在伺服器上訓練並保存模型,複製到本地之後load_model()報錯: ValueError: Tensor conversion requested dtype int32 for Tensor with dtype float32: 'Tensor("embedding_1/random_uni ...
在伺服器上訓練並保存模型,複製到本地之後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 指定版本安裝:
$ pip install Keras==2.0.3