Tensorboard callback does not show model_stats and image

Hi,
Anyone has suceeded using tensorboard callback of fastai with DynamicUnet on google colab?
I am using LearnerTensorboardWriter() and it only records metric values, failed to record any model_stats values.
The error message pop up is:
IOPub data rate exceeded.
The notebook server will temporarily stop sending output
to the client in order to avoid crashing it.
To change this limit, set the config variable
--NotebookApp.iopub_data_rate_limit.

Current values:
NotebookApp.iopub_data_rate_limit=1000000.0 (bytes/sec)
NotebookApp.rate_limit_window=3.0 (secs)