Le callback Keras Tensorboard n'écrit pas les images

Question

J'essaie de visualiser les poids de mon modèle Keras avec Tensorboard. Voici le modèle que j'utilise :

model = Sequential([
    Conv2D(filters=32, kernel_size=(3,3), padding="same", activation='relu', input_shape=(40,40,3)),
    MaxPooling2D(pool_size=(2, 2)),
    Conv2D(filters=64, kernel_size=(5,5), padding="same", activation='relu'),
    MaxPooling2D(pool_size=(2, 2)),
    Flatten(),
    Dense(1024, activation='relu'),
    Dropout(0.5),
    Dense(43, activation='softmax'),
])
model.compile(optimizer='sgd',
              loss='categorical_crossentropy',
              metrics=['accuracy'])

et je m'entraîne avec cet appel :

model.fit_generator(
    ...
    callbacks = [
        ModelCheckpoint('models/gtsrb1-{epoch}.hdf5', verbose=1, save_best_only = True),
        TensorBoard(log_dir='tblogs/', write_graph=True, write_grads=True, write_images=True),
        EarlyStopping(patience=5, verbose=1),
    ],)

Cependant, lorsque je démarre TensorBoard, voici ce que j'obtiens :

Scalars and Graphs semble correct, il ne s'agit donc pas d'un problème d'erreur. logdir . Qu'est-ce que je fais de travers ?

Answer 1

Vous devez ajouter histogram_freq=x où x doit être différent de zéro, afin que l'écriture des images soit activée.

Mais si vous faites cela, cela peut encore échouer, en fonction de la version de Keras (voir https://github.com/fchollet/keras/issues/6096 )