TPU profiling extension

taomanwai / tensorboardcolab

A library make TensorBoard working in Colab Google

MIT License

149 stars 26 forks source link

TPU profiling extension #5

Open ketyi opened 5 years ago

ketyi commented 5 years ago

TensorBoard startup is extended with TPU profiling option (in case of TPU is present in the Colab environment)

See: https://cloud.google.com/tpu/docs/cloud-tpu-tools

ketyi commented 5 years ago

I think this TPU error has nothing to do with my modifications. In my experience it can be due to compatibility issues between the installed Tensorflow and Keras versions. A solution is to import the Keras pieces from tensorflow.python.keras . I have modified it accordingly, could you give it a try?