Tensorboard on google colab

recipe.py

# imports, etc
from keras.callbacks import TensorBoard

# install ngrok
!wget https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-amd64.zip
!unzip ngrok-stable-linux-amd64.zip

# specify the log location
LOG_DIR = './log'
get_ipython().system_raw(f'tensorboard --logdir {LOG_DIR} --host 0.0.0.0 --port 6006 &')
get_ipython().system_raw('./ngrok http 6006 &')

# get a link to tensorboard session
! curl -s http://localhost:4040/api/tunnels | python3 -c \
    "import sys, json; print(json.load(sys.stdin)['tunnels'][0]['public_url'])"

# preprocess data, create model, etc ...

# run model
RUN_NAME = 'first model' 
callback = TensorBoard(log_dir=f'{LOG_DIR}/{RUN_NAME}')
history = model.fit(... , callbacks=[callback])
# history = model.fit_generator( ... , callbacks = [callback]) , etc