使用tensorboard查看网络结构

在tf2.0中,可以添加如下代码:

    log_dir='./log/'
    tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)
    history = classifier.fit(data_train, label_train, validation_data=(data_test, label_test), epochs=epochs,
                             batch_size=64,callbacks=[tensorboard_callback])
    .....

然后,tensorboard --logdir=log就可以在浏览器中打开tensorboard了。