2017-10-14 8 views
0

a==bの場合、次のコードが失敗するのはなぜですか? エラー:私はバックエンドとkerasバージョン2.0.1としてPython 3、TensorFlow 1.1.0を使用 ValueError: Shapes must be equal rank, but are 2 and 1 for 'Assign' (op: 'Assign') with input shapes: [12,12], [12].ケラスでローディングウェイトが失敗する

#!/usr/bin/env python3 
import keras 

a = 12 
b = 12 
x = keras.layers.Input(shape=(a,)) 
y = keras.layers.Dense(b)(x) 
model = keras.models.Model(inputs=[x], outputs=y) 

model.save_weights("model.h5") 
model.load_weights("model.h5") 

。 OSはLinuxです。

+2

は、TensorFlow 1.3.0、Keras 2.0.8で動作します –

答えて

0

Tensorflow 1.3.0にアップグレードしても問題が解決されない Keras 2.0.8にアップグレードすると、問題が解決されました。

明らかに、読み込みでは保存するのではなく問題が発生しました。

関連する問題