Source: Deep Learning on Medium
With the recent interest in Artificial Intelligence, Machine Learning and Data Analytics, one of the biggest hurdles for Machine Learning Engineer, Data Scientist and Data Analyst is allocate the right hardware for model training. However, now it is possible for anyone to kick off model training with Google Colab.
Google Colab is a free to use Jupyter notebook, that allows you to use free Nvidia Tesla T4 GPUs and 12 GB of RAM which you can use it up to 12 hours in row.
With TensorFlow 2.0, it is possible to visualize your TensorFlow graph, plot quantitative metrics about the execution of your graph, and show additional data like images that pass through it using TensorBoard directly from within Google Colab.
#install TensorFlow 2.0
!pip install tensorflow==2.0.0-alpha0
# Load the TensorBoard notebook extension
Add to tf.keras callback
tensorboard_callback = tf.keras.callbacks.TensorBoard(logdir, histogram_freq=1)
Start TensorBoard within the notebook using magics function
%tensorboard — logdir logs
Now you can view your TensorBoard from within Google Colab.
Full source code can be downloaded from here