Le chargement des poids échoue dans les keras

Question

Pourquoi le code suivant échoue-t-il si a==b, mais fonctionne correctement dans le cas contraire? Erreur: 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") 

J'utilise python 3, tensorflow 1.1.0 comme back-end et KERAS la version 2.0.1. OS est un peu de Linux.

Answer 1

Mise à niveau vers Tensorflow 1.3.0 n'a pas résolu le problème Cependant, mise à niveau vers Keras 2.0.8 a résolu le problème.

Apparemment, le chargement a causé le problème plutôt que de sauver.