GOOGLE COLAB 之TENSORBOARD 启用

网上一堆文章，将在GOOGLE COLAB下如何启用tensorboard, 步骤多，不易调通。

下面是简单几步，让我们在colab用起tensorboard:

1.步骤1：确定tensorboard的dir, 如当前目录的 log 下

2.步骤2：在当前会话启用 tensorboard

%load_ext tensorboard

3.步骤三：定义模型训练时的callbacks

 tbCallBack =tf.keras.callbacks.TensorBoard(
                         log_dir=LOG_DIR, 
                         histogram_freq=1,
                         embeddings_freq=1
                         )

history = model.fit(train_dataset, 
                    epochs=5,
                    validation_data=val_dataset, 
                    validation_steps=30,
                    callbacks=tbCallBack)

4.步骤四： 运行并查看tensorboard

%tensorboard --logdir=log